Linux系统 pytorch环境配置-成功了

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

输出为：

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命令下好多包网络连接不上，而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环境。

image.png