下载cuda对应的版本: 安装时需要删除samples等其他几个选项,只安装cuda即可
CUDA Toolkit 11.4 Downloads | NVIDIA Developer
只装Toolkit, cuda
配置bashrc里面的环境:
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
下载cudnn版本:
cuDNN Archive | NVIDIA Developer
解压之后执行下面三条命令:
1.分别输入
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include/
sudo cp cuda/include/cudnn_version.h /usr/local/cuda-11.3/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64