1. 安裝anaconda
安裝包下載:安裝包下載有兩個(gè)途徑却汉,一個(gè)是清華大學(xué)開(kāi)源軟件鏡像站:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
另一個(gè)是自身的https://repo.anaconda.com/archive/index.html (這個(gè)更快一些)盏求。查看Linux系統(tǒng)位數(shù):使用
cat /proc/version
命令可以看出Linux系統(tǒng)的位數(shù)是x86_64的薄嫡。具體anaconda的版本不是十分重要秕脓,因?yàn)榭梢詣?chuàng)建不同的環(huán)境透乾,下載不同的python版本饵筑。本文下載的anaconda版本為Anaconda3-2021.04-Linux-x86_64.sh捧搞,復(fù)制到Linux服務(wù)器上后圾浅,安裝命令為:定位到安裝包所在位置腹尖,輸入如下命令
bash Anaconda3-2021.04-Linux-x86_64.sh
2. 創(chuàng)建指定python版本的虛擬環(huán)境
# 創(chuàng)建
conda create -n gcn python==3.7.3
# 激活
conda activate gcn
同時(shí)確認(rèn)有CUDA環(huán)境柳恐,輸入為
lspci | grep -i vga
輸出為:
image.png
3. 安裝指定版本的pytorch
官方鏈接:https://pytorch.org/get-started/previous-versions/
選定版本的conda命令為:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch
conda命令下好多包網(wǎng)絡(luò)連接不上,而pip的命令為:
pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
4. 測(cè)試安裝成功與否
在pytorch環(huán)境下热幔,輸入python乐设,import torch,然后采用torch.cuda.is_available绎巨,返回結(jié)果為真伤提,即可采用GPU環(huán)境。
image.png