pytorch_geometric离线安装

2022-12-19

字数统计: 411字 | 阅读时长: 2分

× 文章目录

Author: haoransun
WeChat: SHR—97

https://pytorch-geometric.com/
https://pytorch.org/
https://zhuanlan.zhihu.com/p/602615941

cu111 11.1 以上适合实验室3090的GTX算力：8.6

此处采用1.10.0+cu113更合适

1 2	# CUDA 11.3 pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html

安装模板

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

实际安装，替换为适合自己的版本

pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html
pip install 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

已知正确版本对比

进入后下载torch_scatter 、torch_sparse 、torch_cluster 、torch_spline_conv四个包：

python版本是3.8的选cp38，还要判断是win还是linux。

此处是cp37

右键点击，下载到本地

上传到所需服务器上，切换环境，进行安装

pip install torch_cluster-1.5.9-cp37-cp37m-linux_x86_64.whl
pip install torch_scatter-2.0.9-cp37-cp37m-linux_x86_64.whl
pip install torch_sparse-0.6.12-cp37-cp37m-linux_x86_64.whl
pip install torch_spline_conv-1.2.1-cp37-cp37m-linux_x86_64.whl

pip install torch-geometric==2.0.3