Linux服务器配置Pytorch与Pytorch-geometric

最进更新于：2022/5/2 

 环境说明：

• 环境：ubuntus 16
• 显卡：rtx3090ti
• cuda：11.3
• anaconda：忘了，无所谓
• python:3.6 (虚拟环境）

注：优先尝试Pytorch官网指导的安装指令，去掉最后的-c pytorch用镜像加速

如conda安装遇以下问题才情继续：

问题描述	解决方法
Pytorch官网conda指令安装发现pytorch为cpu版 / 安装时condaHttp000报错	使用官网pip指令安装
Pytorch-geometric官网指令安装并使用时因'GLIBC_2.27'缺失导致报错	不安装pytorch_spline_conv

• conda源中对pytorch各版本的管理可能有问题，在确认安装前需要检查版本信息
• ubuntu 版本低于18时系统缺少关键文件'GLIBC_2.27'，pytorch_spline_conv缺少依赖

终端核心指令：

# 新建一个虚拟环境
conda create -n <name>

# 激活环境
conda activate <name>

# 使用pip安装 torch-1.10.0+cu113 torchaudio-0.10.0+cu113 torchvision-0.11.0+cu113
pip3 install torch==1.10.0+cu113 torchvision==0.11.0+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

# 使用pip安装 torch-geometric
pip install torch-scatter torch-sparse torch-cluster torch-geometric -f https://data.pyg.org/whl/torch-1.10.0+cu113.html

# 如果ubuntu版本>=18，安装 torch-geometric用下面的
pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.10.0+cu113.html