一、找到torch 和 torchvision的对应版本
这里针对python3.9版本,在网站https://github.com/pytorch/vision读者可自行查找python版本对应的torch和torchvision。
二、下载本地CPU版本的torch和torchvision
1.下载链接:https://download.pytorch.org/whl/torch_stable.html
2.选择torch版本
提示:不推荐在线安装torch,但可能因为网络和下载资源问题会有异常报错情况,如果想在线安装可以多试几次pip install torch
pip install torch
3.选择torchvision版本
三、安装
1.打开环境
linux:conda activate 环境名
windows:activate 环境名
2.输入 pip install “文件路径”
torch:
(ComPython39) sxw@sxw-server:/$ pip install /home/sxw/torch-2.1.0+cpu-cp39-cp39-linux_x86_64.whl
torchvision:
(ComPython39) sxw@sxw-server:/$ pip install /home/sxw/torchvision-0.16.0+cpu-cp39-cp39-linux_x86_64.whl
四、测试
1.pip list或conda list 查看版本号
(ComPython39) sxw@sxw-server:/$ pip list
Package Version
------------------ ----------
certifi 2023.7.22
charset-normalizer 3.3.0
filelock 3.12.4
fsspec 2023.10.0
idna 3.4
Jinja2 3.1.2
MarkupSafe 2.1.3
mpmath 1.3.0
networkx 3.2
numpy 1.26.1
Pillow 10.1.0
pip 23.3
requests 2.31.0
setuptools 68.0.0
sympy 1.12
torch 2.1.0+cpu
torchvision 0.16.0+cpu
typing_extensions 4.8.0
urllib3 2.0.7
wheel 0.41.2
2.简单导入测试import,注意print(torch.__version__)的__是两个下划线。
(ComPython39) sxw@sxw-server:/$ python
Python 3.9.0 (default, Nov 15 2020, 14:28:56)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
2.1.0+cpu