在jupyter 运行时出现这样的错误
查看一下,是否是TRUE
Pytorch官网,选好对应版本,
网速快的直接复制粘贴到Ubuntu终端
网速渣的,下个轮子https://pytorch.org/get-started/previous-versions/#pytorch-linux-binaries-compiled-with-cuda-8
进入PyTorch的下载目录,使用pip命令安装:
pip install torch-1.0.0-cp36-cp36m-linux_x86_64.whl
安装torchvision比较简单,可直接使用pip命令安装:
pip install torchvision
最后检查一下
True,ok啦
过程很简单,主要是前期 .whl选错了,再有很蛋疼的就是网速太渣了
涉及的部分博客
https://blog.youkuaiyun.com/red_stone1/article/details/78727096
PS:
查看NVIDIA显卡信息
nvidia-smi
查看CUDA版本
nvcc --version