ubuntu18.04.4安装cuda10.2和cudnn
显卡驱动是上一篇文档中的版本
安装显卡驱动版本cuda-10.2
链接
记得要修改~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
export PATH=$PATH:/usr/local/cuda-10.2/bin
#export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2
export CUDA_HOME=/usr/local/cuda-10.2
卸载cuda方法
种先进入到/usr/local/cuda/bin,再执行对应删除程序呢?如下:
cd /usr/local/cuda-10.2/bin
sudo cuda-uninstaller
赋予文件执行权限
sudo chmod -R 777 cuda_11.1.0_455.23.05_linux.run
安装cudnn
下载对应版本的cudnn
sudo cp include/* /usr/local/cuda-10.2/include/
sudo cp lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
查看cudnn版本(20系列显卡)
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
查看cudnn版本(30系列显卡)cuda11以上
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
显示如下就是代表cudnn版本为7.6.4
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 4
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"