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