其实其他朋友都总结得很详细了,我这里就是稍微做一点补充:
首先按照这里的教程:https://blog.csdn.net/weixin_44776894/article/details/106159483
可以下载并安装Anacoda,装好之后重启终端可以看到前面多了一个(base),输入python或者python3,也都替换成了Anaconda的python。
其次,这个教程介绍了怎样安装CUDA和cuDNN:https://blog.csdn.net/ashome123/article/details/105822040/
首先需要注意的是,在这个页面下选择CUDA版本的时候:https://developer.nvidia.com/cuda-toolkit-archive
保险起见,应该先看看PyTorch所能支持的最新CUDA版本:https://pytorch.org/get-started/locally/
按照上面第二个教程,应该首先安装gcc 7:
sudo apt-get install gcc-7 g++-7
在gcc7安装完成后输入:
ls /usr/bin/gcc*
可以看到目前系统中是有两个gcc的。依次输入:
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50
再输入:
sudo update-alternatives --config gcc
可以看到gcc的默认版本。
之后定位到下载好的CUDA安装包,运行:
sudo sh cuda_10.2.89_440.33.01_linux.run
和上面那个教程不同的地方是,这条命令我运行了两遍(第一次也没有任何报错输出),不过最后总算是运行完了。
另外就是在解压cuDNN的时候,如果用归档管理器打开解压就会报错:Could not create the hard link file:///home/%我的用户名%/cuDNN/cuda/lib64/-libcudnn_static.a
也不知道是什么原因,所以直接运行:
tar zxvf cudnn-10.2-linux-x64-v8.0.2.39.tgz
解压后有个cuda文件,内有include和lib64两个文件夹,进入include文件夹,执行如下命令:
sudo cp cudnn.h /usr/local/cuda/include/
再cd命令切换进lib64文件夹,执行如下命令:
sudo cp lib* /usr/local/cuda/lib64/
最后增加权限:
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
到这一步应该就装好了CUDA和cuDNN了。