辛辛苦苦装系统,搞了两天终于成功。记录一下安装过程。大体上的下面的教程还是对的。但是还是有一些坑在里面。
参考教程如下:Ubuntu 20.04安装CUDA 11.0、cuDNN 8.0.5、PyTorch 1.7.0 - BooTurbo - 博客园
但是需要注意以下地方,先安装显卡驱动,在安装cuda的时候,有一项选择是否安装显卡驱动,如下图:
第一个CUDA Driver 一定把X去掉,不然两个显卡驱动会冲突,导致灰屏,我就是这个原因重新装了系统,才定位出原因。
第二个cuda 版本和pytorch版本一定要对应,我先安装了cuda11.5后来发现pytorch 支持11.3不支持11.5所以重新安装显卡驱动和cuda,浪费很多时间。
第三个需要注意,用conda 安装Pytoch的时候一定要先建立虚拟环境,命令如下:
conda creat -n Pytorch
conda activate Pytorch
在虚拟环境中执行如下命令才能安装成功:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
当然也可以切换清华源。
希望踩过得坑对别人有帮助。