CUDA
卸载
# 卸载 runfile 方式安装的 CUDA
sudo /usr/local/cuda-X.Y/bin/uninstall_cuda_X.Y.pl
# 卸载 Deb 方式安装的 CUDA
sudo apt-get --purge remove cuda
sudo apt-get autoremove
安装
查看显卡驱动
nvidia-smi

# 按序执行下列命令行
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
# 下载 pin 码,若终端无法链接则直接复制上述链接进入网页进行下载即可,已下载好的复制即可
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
# 移动 cuda 的 pin 码
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
# 下载 deb 格式的本地安装包,若终端无法链接则直接复制上述链接进入网页进行下载即可
sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
# 安装
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
# 添加钥匙
sudo apt-get update
不要执行官网的最后一条命令
这里千万注意,不要执行官网的 sudo apt install cuda -y
,因为他会直接获取最新的版本进行安装,会覆盖你上面安装的内容,我们这里只需要安装相对应的版本即可
这里千万注意,不要执行官网的 sudo apt install cuda -y
,因为他会直接获取最新的版本进行安装,会覆盖你上面安装的内容,我们这里只需要安装相对应的版本即可
这里千万注意,不要执行官网的 sudo apt install cuda -y
,因为他会直接获取最新的版本进行安装,会覆盖你上面安装的内容,我们这里只需要安装相对应的版本即可
这里记得安装修改版本号
这里记得安装修改版本号
这里记得安装修改版本号
sudo apt install cuda-toolkit-10-2 -y
配置环境配置
在sudo gedit ~/.bashrc
添加
注意版本号
注意版本号
注意版本号
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib:$LD_LIBRARY_PATH
更新变量
source ~/.bashrc
检查
nvcc -V
CUDNN
卸载
# cudnn.h 和 libcudnn*
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
sudo rm -rf /usr/include/cudnn.h
sudo rm -rf /usr/lib/x86_64-linux-gnu/libcudnn*
安装

- 安装这三个文件
- 将
/usr/include/cudnn.h
复制到/usr/local/cuda/include
- 将
/usr/lib/x86_64-linux-gnu/libcudnn*
复制到/usr/local/cuda/lib64
验证
cp -r /usr/src/cudnn_samples_v7/ $HOME
(根据自己的版本号)cd cudnn_samples_v7/mnistCUDNN/
(根据自己的版本号)make clean &&make
./mnistCUDNN
(不成功则重启再试一次即可)
错误解决
error:FreeImage.h 没有这个文件或目录
原因:缺少相关的依赖文件包
sudo apt-get install libfreeimage3 libfreeimage-dev