本文共 274 字,大约阅读时间需要 1 分钟。
本文的安装过程适用于在Ubuntu下已经安装好GPU和conda的计算机。
不翻墙装软件会报错。
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
$ jupyter notebook
在.ipynb中输入:
import torchimport torchvisionprint(torch.cuda.is_avaiable())
显示输出:
True[1]
[2]转载地址:http://vlzgi.baihongyu.com/