机器学习_Python对应版本Pytorch及PyG安装_python与torch版本

作者：weixin_40725706 | 2024-07-24 22:30:27

踩

python与torch版本

一、torch\torchaudio\torchvision安装

1. python版本与torch版本对应

下表来自pytorch的gituhub官网

`torch`	`torchvision`	Python
`main` / `nightly`	`main` / `nightly`	`>=3.8`, `<=3.12`
`2.3`	`0.18`	`>=3.8`, `<=3.12`
`2.2`	`0.17`	`>=3.8`, `<=3.11`
`2.1`	`0.16`	`>=3.8`, `<=3.11`
`2.0`	`0.15`	`>=3.8`, `<=3.11`

older versions

`torch`	`torchvision`	Python
`1.13`	`0.14`	`>=3.7.2`, `<=3.10`
`1.12`	`0.13`	`>=3.7`, `<=3.10`
`1.11`	`0.12`	`>=3.7`, `<=3.10`
`1.10`	`0.11`	`>=3.6`, `<=3.9`
`1.9`	`0.10`	`>=3.6`, `<=3.9`
`1.8`	`0.9`	`>=3.6`, `<=3.9`
`1.7`	`0.8`	`>=3.6`, `<=3.9`
`1.6`	`0.7`	`>=3.6`, `<=3.8`
`1.5`	`0.6`	`>=3.5`, `<=3.8`
`1.4`	`0.5`	`==2.7`, `>=3.5`, `<=3.8`
`1.3`	`0.4.2` / `0.4.3`	`==2.7`, `>=3.5`, `<=3.7`
`1.2`	`0.4.1`	`==2.7`, `>=3.5`, `<=3.7`
`1.1`	`0.3`	`==2.7`, `>=3.5`, `<=3.7`
`<=1.0`	`0.2`	`==2.7`, `>=3.5`, `<=3.7`

2. python与torch对应版本安装

环境：- Vmware Ubuntu16.04 or Ubuntu20.04 - Anaconda python=3.8
1.torch1.13+cpu 版本
2.torch1.13+cuda 版本

# torch1.13+cpu 版本
pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --index-url https://download.pytorch.org/whl/cpu
1
2

https://download.pytorch.org/whl/cpu为torch1.13.1+cpu版本

https://download.pytorch.org/whl/cu116为torch1.13.1+cuda116版本

二、torch_geometric安装

PyG官网安装

1. torch_geometric可选依赖包安装

环境：- Vmware Ubuntu16.04 or Ubuntu20.04 - Anaconda python=3.8
1.torch1.13+cpu 版本
2.torch1.13+cuda 版本

# Optional dependencies:
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-1.13.1%2Bcpu.html
1
2

https://data.pyg.org/whl/torch-1.13.1%2Bcpu.html为torch1.13.1+cpu版本对应的torch_geometric依赖包。
https://data.pyg.org/whl/torch-1.13.1%2Bcu116.html为torch1.13.1+cuda11.6版本对应的torch_geometric依赖包。
其他torch对应版本可在https://data.pyg.org/whl中寻找。

2. torch_geometric安装

pip install torch_geometric
1

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop】