1、介绍
显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么? - 知乎
2、装cuda
CUDA Toolkit 10.1 original Archive | NVIDIA Developer
3、环境变量
CUDA和cudnn的环境变量设置问题_zl535320706的博客-CSDN博客_nvcc环境变量
4、装cudnn
cudnn的安装非常简单,
(1)下载安装文件
按需求下载cudnn的安装文件:cuDNN Archive | NVIDIA Developer
(2)安装cudnn
解压下载的文件,可以看到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*
(3)查看cudnn版本
在终端输入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出现下图所示版本信息,说明安装成功。