也是弄了一下午,但是好像是他自己装好的。
首先是从上面的网站上下一个适合的驱动,然后通过
sudo systemctl set-default multi-user.target
关闭桌面
然后sh 安装之后开启桌面
sudo systemctl set-default graphical.target
但是其实起作用的是
sudo apt install nvidia-cuda-toolkit这个命令
安装完之后nvcc -V 看一下装的是什么
然后conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/添加清华源
然后conda install pytorch1.0.1 torchvision0.2.2 cudatoolkit=9.0 安装