Ubuntu 18.04 深度学习环境快速配置命令记录

1、更新系统包

sudo apt-get update

sudo apt-get upgrade

2、安装显卡驱动

sudo apt-get install ubuntu-drivers-common   #安装ubuntu-drivers

sudo ubuntu-drivers autoinstall

#查看显卡安装版本

ubuntu-drivers devices

4、在线安装不行，离线安装ok,不过离线安装需要安装gcc，

sudo apt install build-essential

bash XXX.run

470下载地址

5、安装miniconda

Miniconda — conda documentation

6、初始化conda

7、安装cuda

CUDA Toolkit Archive | NVIDIA Developer

貌似需要在线的命令比较顺畅，下面命令不行的话，在上面链接找对应版本。

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin

sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget https://developer.download.nvidia.com/compute/cuda/11.4.3/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.3-470.82.01-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.3-470.82.01-1_amd64.deb

sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub

sudo apt-get update

sudo apt-get -y install cuda

8、source ~/.bashrc

9、安装torch

pip install torch==1.8.0 torchvision==0.9.1

离线安装torch

离线包地址

 torch 1.8.0 +cu111       vision 0.9.0+cu111

10、安装cudann

cuDNN Archive | NVIDIA Developer

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.4/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.4/lib64

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

sudo dpkg -i libcudnn8_8.2.4.15-1+cuda11.4_amd64.deb
sudo dpkg -i libcudnn8-dev_8.2.4.15-1+cuda11.4_amd64.deb
sudo dpkg -i libcudnn8-samples_8.2.4.15-1+cuda11.4_amd64.deb

11、配置环境变量

export PATH="/usr/local/cuda-11.4/bin${PATH:+:${PATH}}"
export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}"

source ~/.bashrc

12、conda换镜像源

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes