安装NVIDIA,CUDA, cuDNN的个人笔记

发布时间 2023-10-13 11:08:25作者: 独上兰舟1

安装nvdian:

sudo chmod +x NVIDIA-Linux-X86_64-440.100.run

sudo ./NVIDIA-Linux-X86_64-440.100.run -no-x-check

在后面加上不对Xserver进行检查的命令(红色)就可以安装成功!

 

 查看INVIDIA

lspci -v -s `lspci | awk '/ VGA / {print $1}'`

 

 

安装cuda网站:https://developer.nvidia.com/cuda-toolkit-archive

注:**此处以将cuda安装至software文件夹中为例,完成下述安装步骤说明:
(1)

sudo ./cuda_10.2.89_440.33.01_linux.run
或者
sudo sh ./cuda_10.2.89_440.33.01_linux.run

检查cuda是否安装成功:nvcc -V

如果没有这个命令的话,需要配置。

vim ~/.bashrc

在文件末尾添加(这里的/usr/local/cuda-11.3替换成你的cuda安装路径,一般都在/usr/local路径下)

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.3/lib64
export PATH=$PATH:/usr/local/cuda-11.3/bin
# export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.3
export CUDA_HOME=/usr/local/cuda-11.3

 

2.

安装cudnn网站:https://developer.nvidia.com/rdp/cudnn-archive

 

sudo cp include/cudnn.h /usr/local/cuda-11.7/include


sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64

sudo chmod a+r /usr/local/cuda-11.7/include/cudnn.h /usr/local/cuda-11.7/lib64/libcudnn*
查看cudnn版本:

cat /usr/local/**cuda-11.x**/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

 

3.首先是判断cuda是否安装成功。

一般安装路径为/usr/local/cuda

使用nvcc -v命令可以输出cuda版本

然后是判断cudnn,这个库安装很简单,只需把cudnn的include和lib64里面的文件拷到cuda相应目录即可,所以判断是否安装的方式是,到cuda的include和lib64,用ls | grep cudnn 命令查看是否有cudnn相关的文件。