ubuntu18.04安装cuda

卸载之前安装的驱动

sudo apt-get purge nvidia*

安装驱动

• 方法1：

查看显卡适配的NVIDIA的驱动

ubuntu-drivers devices

安装recommend推荐的版本

sudo apt-get install nvidia-driver-455

• 方法2：

或者直接使用下面命令安装

sudo ubuntu-drivers autoinstall

安装完毕后，需要重启

sudo reboot

安装CUDA

CUDA官网

sudo sh cuda_10.1.105_418.39_linux.run

accept后，将驱动（Driver）取消后，选择Install

安装成功

添加环境变量

sudo vim ~/.bashrc

// 添加下面三行
export CUDA_HOME=/usr/local/cuda-10.1
export PATH=/usr/local/cuda-10.1/bin${PATH:+:$PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc

测试安装是否成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

安装CUDNN

CUDNN

下载cuDNN Library for Linux，在解压的文件夹中打开终端，文件夹中有cuda文件夹

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

测试

import torch
torch.cuda.is_available()

查看当前cudnn的版本

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

卸载

卸载CUDA很简单，一条命令就可以了，主要执行的是CUDA自带的卸载脚本，读者要根据自己的cuda版本找到卸载脚本：

sudo /usr/local/cuda-10.1/bin/uninstall_cuda_10.1.pl
sudo rm -rf /usr/local/cuda-10.1/