1.安装cuda
英伟达官网其实有非常纤细的教程,如果装新版cuda建议直接看官网
不要找cuda9cuda8的旧教程。。。(虽然其实差不多)
安装的时候注意安装选项即可,比如装cuda的时候不要勾选安装nvidia驱动
建议选择使用 .run
文件安装,因为使用 .deb
可能会将已经安装的较新的显卡驱动替换。
2.安装cudnn的命令
tar -xvf cudnn-11.4-linux-x64-v8.2.4.15.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.4/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.4/lib64
sudo chmod a+r /usr/local/cuda-11.4/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.4/lib64/libcudnn*
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
sudo chmod a+r /usr/local/cuda-11.3/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.3/lib64/libcudnn*
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*
4.查看cuda版本及软链接
查看/usr/local里面 cuda 版本的软链接——cuda
进入/usr/local目录 使用 stat cuda命令
3. cuda版本切换
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.4 /usr/local/cuda
一定要注意,是cuda,不是cuda/,否则emmmm
cuda,cudnn官网,官网会有旧版本选择,耐心找不慌