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简易安装教程