1. 安装anaconda
安装包下载:安装包下载有两个途径,一个是清华大学开源软件镜像站:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
另一个是自身的https://repo.anaconda.com/archive/index.html (这个更快一些)。查看Linux系统位数:使用
cat /proc/version
命令可以看出Linux系统的位数是x86_64的。具体anaconda的版本不是十分重要,因为可以创建不同的环境,下载不同的python版本。本文下载的anaconda版本为Anaconda3-2021.04-Linux-x86_64.sh,复制到Linux服务器上后,安装命令为:定位到安装包所在位置,输入如下命令
bash Anaconda3-2021.04-Linux-x86_64.sh
2. 创建指定python版本的虚拟环境
# 创建
conda create -n gcn python==3.7.3
# 激活
conda activate gcn
同时确认有CUDA环境,输入为
lspci | grep -i vga
输出为:
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命令下好多包网络连接不上,而pip的命令为:
pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
4. 测试安装成功与否
在pytorch环境下,输入python,import torch,然后采用torch.cuda.is_available,返回结果为真,即可采用GPU环境。