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!