背景描述:
环境: ubuntu16.04+cuda8.0+pytorch0.4.1+anaconda3
想要另外安装一个pytorch1.0.0
步骤:
1.在anaconda3中另外建一个环境:conda create [name]
2.激活环境source activate [name] (退出环境用source deactivate)
3.查看cuda 以及显卡驱动是否完好
nvcc -V查看cuda, nvidia-smi查看显卡驱动
4.没问题就安装pytorch1.0.0: conda install pytorch=1.0.0 cuda80 -c pytorch
接着安装torchvision
5.如果安装后import torch 没问题,但是torch.cuda.is_available()有问题返回的是false
6.重启电脑试试!!!(重启电脑很重要,以及重启终端,有时候忙活半天问题就在这里)