1.cuda安装

选择对应版本,系统,下载runfile{local}文件。

https://developer.nvidia.com/cuda-toolkit-archive

sudo sh cuda_10.2.89_440.33.01_linux.run#对应文件名称

驱动去除(之前安装过)

另一种安装方式,关闭图像界面lightdm进入命令行ctrl+alt+f1安装。

安装完成设置环境

sudo gedit ~/.bashrc
#插入以下内容cuda-x.x为cuda版本号
export  PATH=/usr/local/cuda-x.x/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-x.x/lib64$LD_LIBRARY_PATH
nvcc -V #安装完成查看版本

2.cudnn安装

https://developer.nvidia.com/rdp/cudnn-download下载需注册NVIDIA账号

下载与cuda版本对应的cudnn版本

cudNN Library for Linux

解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令

sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-x.x/lib64/
sudo cp  cuda/include/cudnn.h /usr/local/cuda-x.x/include/

sudo chmod a+r /usr/local/cuda-x.x/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-x.x/lib64/libcudnn*# 为所有用户设置读取权限
sudo apt-get install libcupti-dev #安装libcupti,这是 NVIDIA CUDA 分析工具接口,此库提供高级分析支持.

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

3.cuda卸载

sudo /usr/local/cuda-x.x/bin/cuda-uninstaller

sudo rm -rf /usr/local/cuda-x.x/

4.cudnn卸载

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn