Ubuntu16.04下基于Conda安装PyTorch

本文详细介绍了如何在Ubuntu16.04系统中,使用Conda环境安装PyTorch及其依赖项,包括翻墙下载组件的方法和验证安装是否成功的步骤。适合已配置好GPU和conda的计算机。

Ubuntu16.04下基于Conda安装PyTorch

安装环境

本文的安装过程适用于在Ubuntu下已经安装好GPU和conda的计算机。

翻墙

不翻墙装软件会报错。

命令行安装

在这里插入图片描述

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

安装验证

$ jupyter notebook

在.ipynb中输入:

import torch
import torchvision
print(torch.cuda.is_avaiable())

显示输出:
True

参考资料

[1] PyTorch官方
[2] Ubuntu16.04环境下PyTorch简易安装教程

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值