cuda-cuDnn
cuda
sudo /bin/sh cuda_11.7.0_515.43.04_linux.run
cudnn
cuDNN Archive | NVIDIA Developer
Linux 系统 CUDA 多版本共存以及切换 – 颢天
安装cuda
# 如果已经安装过驱动，驱动不需要再安装，取消勾选
安装cuDNN，cuda-cuDNN对应关系见上，相应的从上面的下载，
解压
tar -xvf xxx.tar.xz
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda-12.1/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda-12.1/lib64
sudo chmod a+r /usr/local/cuda-12.1/include/cudnn*.h /usr/local/cuda-12.1/lib64/libcudnn*
vim ~/.bashrc
文件尾部添加下面内容
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
function switch_cuda {
if [[ $1 =~ ^[0-9]+\.[0-9]+$ ]]; then
sudo rm /usr/local/cuda
sudo ln -s /usr/local/cuda-$1 /usr/local/cuda
nvcc --version
else
echo invalid version
fi
}
source .bashrc