CUDA and CUDNN
CUDA and CUDNN

CUDA and CUDNN

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

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注