- 下载安装包
wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run
- 执行安装命令
sudo sh cuda_11.2.0_460.27.04_linux.run
输入命令后，会出现如下的界面，这个时候选择continue
这里输入accept
将光标移动到Driver选项并回车，取消选中后install：
安装完成后，显示如下：
最后不要忘记配置cuda的环境变量
vim ~/.bashrc #按o进入编辑模式，并在最后一行添加如下代码 export PATH=$PATH:/usr/local/cuda-11.0/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64 export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib64 #esc后输入wq退出保存
生效更改
source ~/.bashrc
到此安装完成，输入 nvcc -V 验证 显示：
- 首先进入nvidia官网，如下图所示，选择对应的安装包进行下载
- 解压cudnn
根据官网，和CUDA11.1配套的是CUDNN8.1.1，下载下来的是一个tar文件，那么首先需要解压这个文件，解压之后得到一个cuda文件夹，里面有lib64、include还有一个不知道什么用的文档
tar -xvf cudnn-11.2-linux-x64-v8.1.1.33.tgz
- 将解压文件放到指定位置
将解压缩的文件放入对应cuda项目下，并设上只读权限。
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn*.h sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4.验证
查看cudnn版本信息：
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
如下图所示，可以看到已经安装成功cudnn8.1.1版本
