下载安装CUDA
卸载掉原有驱动:sudo apt-get remove nvidia-*
删除旧内核:sudo apt autoremove
安装NVIDIA显卡驱动:sudo apt install nvidia-driver-525
官网下载:https://developer.nvidia.com/cuda-toolkit-archive
下载安装CUDNN
官网下载:https://developer.nvidia.com/rdp/cudnn-download
自行选择与您下载的CUDA对应的适合的版本
加入环境变量
解压命令:tar -xf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
注意,名字替换为你下载的版本
进入到cudnn的下载目录,然后输入以下代码,将解压后的文件拷贝到CUDA对应的安装目录下,并更改文件属性:
1 sudo mkdir -p /usr/local/cuda/include
sudo mkdir -p /usr/local/cuda/lib64
1、sudo cp include/cudnn.h /usr/local/cuda/include
2、sudo cp lib/libcudnn* /usr/local/cuda/lib64*
3、sudo chmod a+r /usr/local/cuda/include/cudnn.h
4、sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
把cuda文件夹下的bin目录添加到path中
5 export PATH=$PATH:/usr/local/cuda/bin
6 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
要使环境变量修改生效,命令行输入:source ~/.bashrc
可以在终端上查看路径有没有添加进去
echo $PATH
echo $LD_LIBRARY_PATH
查看cuda有没有安装好:
到安装的cuda的bin目录下,打开终端输入命令:nvcc -V
然后sudo reboot
重启即可~