Ubuntu22.04深度学习环境配置

1、前置条件

x86_64的PC机或服务器，安装有NVIDIA显卡，安装Ubuntu 22.04操作系统。

以下操作在本地shell或远程ssh均可。

2、安装显卡驱动（限系统管理员操作）

NVIDIA的驱动作为闭源软件，在RHEL、CentOS、Debian等Linux操作系统的软件包源里是没有的。需要去https://www.nvidia.cn/Download/index.aspx?lang=cn 上自行下载显卡驱动。安装前需要安装gcc等编译相关软件。

下载后$ sudo bash NVIDIA-Linux-x86_64-535.146.02.run

但是Ubuntu软件源中收录NVIDIA驱动，可直接安装。软件源配置文件 /etc/apt/sources.list，可以参考 https://mirrors.nju.edu.cn/mirrorz-help/ubuntu/?mirror=NJU 进行配置。

$ sudo apt update

$ sudo apt install nvidia-driver-535

安装显卡驱动后可以查看是否正常驱动显卡。建议重启系统。

$ nvidia-smi

3、安装anaconda的Python环境

下载地址：https://mirrors.nju.edu.cn/anaconda/ 或官网https://www.anaconda.com/。

自行寻找最新的或者合适的版本。下面以miniconda为例。

$ wget https://mirrors.nju.edu.cn/anaconda/miniconda/Miniconda3-py311_23.11.0-2-Linux-x86_64.sh

$ bash Miniconda3-py311_23.11.0-2-Linux-x86_64.sh

接受默认设置完成安装，然后重载环境，测试是否安装完备。

$ source ~/.bashrc

$ conda -V

4、安装Pytorch

在安装之前配置anaconda的源地址，参考https://mirrors.nju.edu.cn/mirrorz-help/anaconda/?mirror=NJU。

在配置好 ~/.condarc 环境文件之后，参考 https://pytorch.org/get-started/locally/ ,安装pytorch：

$ conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

$ conda install ipython

$ ipython

>>> import torch

>>> torch.cuda.is_available()

输出为True表示安装完好。done!