Linux/Ubuntu系统如何安装cudnn?(适用于所有40系显卡 4090 4080 4070)

下载安装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 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 重启即可~

你可能感兴趣的:(linux,ubuntu,服务器,python,cudnn)