Ubuntu18.04+CUDA10.2 深度学习开发环境配置指南

本机系统配置为Ubuntu18.04+CUDA10.2，显卡信息为GeForce GTX 1080Ti。

1.安装Nvidia显卡驱动

查看本机显卡能够配置的驱动信息：

ubuntu-drivers devices

 

可以看到本机推荐的驱动版本为440，然后根据Nvidia官网上CUDA10.2 的配置要求看一下是否匹配：

官方给出的CUDA10.2最低驱动版本为440.33，所以要安装CUDA10.2的话，驱动器版本只有440可选。依次执行下述命令安装440驱动：

sudo apt-get purge nvidia* sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo apt-get install nvidia-driver-440

然后重启机器使更新生效：

sudo shutdown -r now

2.安装CUDA10.2

 

第二步就是安装CUDA。这里我们安装CUDA最新的10.2版本。进入Nvidia cuda下载地址：

https://developer.nvidia.com/cuda-downloads

 

 依次选择对应系统和版本等信息：

我们这里选择的Ubuntu 18.04下的deb安装类型。然后依次执行下述7条命令：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pinsudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600wget http://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.debsudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.debsudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pubsudo apt-get updatesudo apt-get -y install cuda

如果顺利的话CUDA10.2可以安装完成。但如果本机之前的CUDA版本存在卸载不干净等问题的话，安装有可能会报dkpg的错：

 

所以这里补充一个CUDA完全卸载的方法：​​​​​​

sudo apt-get --purge remove "*cublas*" "cuda*"sudo apt-get --purge remove "*nvidia*"sudo apt-get purge nvidia*sudo apt-get autoremovesudo apt-get autocleansudo rm -rf /usr/local/cuda*

3.安装Python3.7 并配置深度学习环境

下载anaconda：

wget https://repo.anaconda.com/archive/Anaconda3-2020.02-Linux-x86_64.sh

安装：

bash Anaconda3-2020.02-Linux-x86_64.sh

创建虚拟环境：

conda install -n 虚拟环境名称

启动虚拟环境：

conda activate 虚拟环境名称

安装深度学习框架，以PyTorch1.5为例：

 

pip install torch torchvision

安装完成后验证一下当前环境下的PyTorch是否可以使用GPU，显示为True则表明配置成功。

import torchtorch.cuda.is_available()