安装cuda
1、输入命令sudo bash cuda_11.2.0_460.27.04_linux.run
出现界面,点击continue,在选是否安装驱动得时候X掉驱动安装,点一下回车可以X掉,再点回车,等待安装完成,安装完成后添加路径到环境变量中。
vim ~/.bashrc
export PATH=/usr/local/cuda-11.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64:$ LD_LIBRARY_PATH
输入source ~/.bashrc 激活环境变量
输入nvcc -V查看是否安装cuda
安装cudnn
解压cudnn-11.2-linux-x64-v8.1.0.77.tgz
tar -xzvf cudnn-11.2-linux-x64-v8.1.0.77.tgz #解压
sudo cp cuda/include/* /usr/local/cuda-11.2/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.2/lib64/
sudo chmod a+r /usr/local/cuda-11.2/include/*
sudo chmod a+r /usr/local/cuda-11.2/lib64/libcudnn*
安装完成后输入cat /usr/local/cuda-11.2/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
查看cudnn是否安装成功。