選擇帶GPU的鏡像后
更新源，安裝必要依賴

sudo apt update
sudo apt install python3-pip python3-dev python3-tk
sudo apt install python3-yaml
sudo apt install cmake

配置中文環境

 sudo apt-get install language-pack-zh-hans

sudo vim /etc/environment
添加下面兩行

LANG="zh_CN.UTF-8"
LANGUAGE="zh_CN:zh:en_US:en"

接著

sudo dpkg-reconfigure locales

pip3

pip3 install numpy
pip3 install pillow
pip3 install matplotlib

git

git clone https://github.com/pytorch/pytorch.git
git clone https://github.com/pytorch/vision.git

安裝pytorch

cd pytorch
git submodule update --init
python3 setup.py install

安裝vision

cd vision
python3 setup.py install

配置zsh

sudo apt install zsh

個人配置

git clone https://github.com/qianhaoq/pytorch_test.git