安装PyTorch框架
x86_64架构
方式一:
使用以下命令可以直接从PyTorch官方预编译包的下载页面中,选择与本地环境匹配的版本包来安装PyTorch,包括匹配操作系统平台(例如Linux)和匹配系统的Python版本号(例如cp37)。例如torch==1.11.0+cpu:指定安装torch版本为1.11.0,并且是CPU版本。
- 安装1.11.0版本
pip3 install torch==1.11.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
- 安装2.1.0版本
pip3 install torch==2.1.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
- 安装2.2.0版本
pip3 install torch==2.2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
方式二:
- 点击下方PyTorch官方链接下载whl包。PyTorch安装包版本需与Python版本一一对应,例如Python版本为3.8.x,需下载"torch-version+cpu-cp38-cp38-linux_x86_64.whl"PyTorch安装包。
表1 whl包字段说明 字段
描述
torch
表示PyTorch。
version
PyTorch的版本号,例如1.11.0。
cpu
说明是CPU版本的PyTorch。
cp38-cp38
适用于Python 3.8的版本。
linux_x86_64
适用于linux系统的x86_64架构。
whl
Python编译好的wheel二进制发布包格式。
更多PyTorch版本可单击Link进行查询。表2 x86_64常用下载链接 PyTorch版本
Python版本
下载链接
1.11.0
Python 3.7和Python 3.7m
https://download.pytorch.org/whl/cpu/torch-1.11.0%2Bcpu-cp37-cp37m-linux_x86_64.whl
Python 3.8
https://download.pytorch.org/whl/cpu/torch-1.11.0%2Bcpu-cp38-cp38-linux_x86_64.whl
Python 3.9
https://download.pytorch.org/whl/cpu/torch-1.11.0%2Bcpu-cp39-cp39-linux_x86_64.whl
Python 3.10
https://download.pytorch.org/whl/cpu/torch-1.11.0%2Bcpu-cp310-cp310-linux_x86_64.whl
2.1.0
Python 3.8
https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp38-cp38-linux_x86_64.whl
Python 3.9
https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp39-cp39-linux_x86_64.whl
Python 3.10
https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp310-cp310-linux_x86_64.whl
2.2.0
Python 3.8
https://download.pytorch.org/whl/cpu/torch-2.2.0%2Bcpu-cp38-cp38-linux_x86_64.whl
Python 3.9
https://download.pytorch.org/whl/cpu/torch-2.2.0%2Bcpu-cp39-cp39-linux_x86_64.whl
Python 3.10
https://download.pytorch.org/whl/cpu/torch-2.2.0%2Bcpu-cp310-cp310-linux_x86_64.whl
- 执行以下命令安装。以安装适配Python 3.8的PyTorch 1.11.0版本为例,其他版本请根据下载的软件包修改命令:
pip3 install torch-1.11.0+cpu-cp38-cp38-linux_x86_64.whl
如果使用非root用户安装,需要在命令后加--user。
aarch64
- 获取安装包。PyTorch安装包版本需与Python版本一一对应,例如Python版本为3.8.x,需下载"torch-version-cp38-cp38-manylinux2014_aarch64.whl"PyTorch安装包。
表3 whl包字段说明 字段
描述
torch
表示PyTorch。
version
PyTorch的版本号,例如1.11.0。
cp38-cp38
表示这个安装包是为Python 3.8编译的。
manylinux2014_aarch64
表示支持manylinux2014标准的aarch64架构系统(64位ARM)。
whl
表明这是一个Python wheel格式的二进制发布包。
更多PyTorch版本可单击Link进行查询。表4 aarch64常用下载链接 PyTorch版本
Python版本
下载链接
1.11.0
Python 3.7和Python 3.7m
https://download.pytorch.org/whl/torch-1.11.0-cp37-cp37m-manylinux2014_aarch64.whl
Python 3.8
https://download.pytorch.org/whl/torch-1.11.0-cp38-cp38-manylinux2014_aarch64.whl
Python 3.9
https://download.pytorch.org/whl/torch-1.11.0-cp39-cp39-manylinux2014_aarch64.whl
Python 3.10
https://download.pytorch.org/whl/torch-1.11.0-cp310-cp310-manylinux2014_aarch64.whl
2.1.0
Python 3.8
https://download.pytorch.org/whl/cpu/torch-2.1.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Python 3.9
https://download.pytorch.org/whl/cpu/torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Python 3.10
https://download.pytorch.org/whl/cpu/torch-2.1.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
2.2.0
Python 3.8
https://download.pytorch.org/whl/cpu/torch-2.2.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Python 3.9
https://download.pytorch.org/whl/cpu/torch-2.2.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Python 3.10
https://download.pytorch.org/whl/cpu/torch-2.2.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
用户可使用上方链接直接下载。或者使用wget命令下载,以下载适配Python 3.8的的PyTorch 1.11.0版本为例,执行如下安装命令(其他版本请根据下载的软件包修改命令):wget https://download.pytorch.org/whl/torch-1.11.0-cp38-cp38-manylinux2014_aarch64.whl
- 执行如下命令安装。以安装适配Python 3.8的的PyTorch 1.11.0版本为例,执行如下安装命令(其他版本请根据下载的软件包修改命令):
pip3 install torch-1.11.0-cp38-cp38-manylinux2014_aarch64.whl
如果使用非root用户安装,需要在命令后加--user。
父主题: 安装步骤