在conda环境下安装PyTorch+PyTorch Geometric（Pyg）

关于cuda和cudnn

不需要按网上教程去官网下载，也不需要手动在conda环境内安装。直接按PyTorch官网命令安装PyTorch即可，cuda和cudnn会自动按你选择的版本下载。

建立conda环境并激活

大家应该都会

conda create -n 你的环境名字 python=你想要的python版本
如：
conda create -n pytorch python=3.6

激活

conda activate 你的环境名字

安装PyTorch

官网链接：Previous PyTorch Versions | PyTorch

如果不需要特定版本，目前比较稳定的版本可选pytorch 1.8.0 +CUDA 10.2。激活环境后直接运行下面代码即可：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

其他版本按官网指示来，需要改动几个数字

重点：不要安装最新版和带有".1"的版本，比如不要装 1.8.1（大坑！）。因为后面的Pytorch geometric （Pyg）兼容性不够好，会出各种问题。

更新：2022/4/26，上述命令不能用了，建议大家去官网使用最新版的指令

测试

python
import torch
print(torch.__version__)
print(torch.cuda.is_available())

返回你的PyTorch版本和Ture既是安装成功

安装Pyg

按CTRL+D退出python，如果上一步选择的PyTorch1.8.0版本（或1.8.0以上），此时直接运行下面代码即可，亲测1.8.0没问题，安装完成

conda install pyg -c pyg -c conda-forge

其他版本按官网指示Installation — pytorch_geometric 2.0.4 documentation

找到对应的pip代码运行即可

Tips

如果之前安装出现问题，建议新建一个环境，不在老环境里修改。新环境第一个就装PyTorch，会自动装需要的其他包。

用conda命令安装时它会检测你环境里缺少的包，然后自动给你一起安装。这很方便，但是有时会装一些不正确的包，导致一些冲突。所以一定要检查它要给你装哪些包，确认没问题在确认安装。