安装环境
本文的安装过程适用于在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简易安装教程