一、安装 cuda
1. 首先查看自身电脑最高支持的cuda版本为多少,在终端输入以下指令
nvidia-smi
可以看到我的最高支持 cuda 11.4 。
2. 打开英伟达官网 下载官方cuda
此处我下载的是cuda 11.0.3 版本,并选择相应配置,复制指令至终端下载
3. 运行官网安装指令
** 4. 依次选择continue和accept 点击确认**
** 5. 按回车取消Driver项的安装,然后回车选择Install**
** 6. 添加环境变量**
gedit ~/.bashrc打开文件,在文件结尾添加如下语句:
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
7. 保存、更新环境变量
source ~/.bashrc
8. 输入nvcc -V查看版本号
二、 安装 cudnn
1. 进入官网 ,寻找合适的 cudnn 版本,下载对应版本的cudnn 的 runtime developer 和samples
2. 下载得到四个文件
3. 安装上述软件包
tar -zxvf cudnn-11.0-linux-x64-v8.0.4.30.tgz
sudo cp -rf cuda/include/cudnn* /usr/local/cuda/include/
sudo cp -rf 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*
sudo dpkg -i libcudnn8_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.4.30-1+cuda11.0_amd64.deb
** 4. 测试**
拷贝例程到用户目录
cp -r /usr/src/cudnn_samples_v8/ $HOME
5. 转到minist例程
cd $HOME/cudnn_samples_v8/mnistCUDNN
6. 编译例程
sudo make clean && make
7. 运行例程
./mnistCUDNN
8. 如果cudnn被恰当的安装,将输出打印消息:Test passed!
9. cudnn 检测版本号
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
版权归原作者 FAST-ROBOT 所有, 如有侵权,请联系我们删除。