注意和之前版本有所不同
1.官网下载cuda 11
2.安装命令
sudo sh cuda_***_linux.run
3.设置环境变量
vi .bashrc #或者 gedit .bashrc
添加以下路径
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
export PATH=$PATH:/usr/local/cuda-11.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.1
4.验证cuda
nvcc -V
5.下载cudnn8,并解压
tar -xvf cudnn-***.tgz
6.复制cudnn8相关文件(需要全部复制include下面的文件)
sudo cp cuda/include/cudnn* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*