Linux 安装Pytorch的指定版本

背景提要:需要借用GitHub的一个代码进行流预测,PyTorch RGCN (Link Prediction)

环境需求:安装如下版本的包
This code is lastly tested with:
python 3.7.x
pytorch 1.7.x
torch_geometric 1.7.x, with torch_scatter 2.0.6 and torch_sparse 0.6.9

1.安装指定版本的pytorch

官方链接:https://pytorch.org/get-started/previous-versions/

从官网查看所需命令:
针对CUDA 10.2版本,所需命令:pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.7.1+cu102.html

安装过程:


安装pytorch
2. 安装指定版本的torch_geometric (方法无效)

官方链接:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

采用Installation via Pip Wheels方法安装

  • 查看pytorch版本和cuda版本,分别是上述的1.7.1和10.2,


    image.png
  • 修改替换得到如下命令
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.7.1+cu102.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.7.1+cu102.html
pip install torch-geometric

以上方法失误

哈哈 后面发现Linux没有GPU,懒得搞CPU版本的了,换成Windows系统重新搞吧

你可能感兴趣的:(Linux 安装Pytorch的指定版本)