1.下载torch和torchvision的安装包
进入torch官网
复制后面的链接并打开,点击torch进入页面,向下滑选择合适的安装包,点击,下载后上传。同样是链接打开的页面往下滑找到torchvision选择合适的安装包,点击,下载后上传。
2.在Linux系统下安装
在终端,同文件的文件夹下执行
pip install torch-xxx.whl
pip install torchvision-xxx.whl --no-deps
使用--no-deps选项表示仅安装torchvision,若不添加会再次安装相关依赖且重新安装 torch。
遇到的问题汇总:
1.ModuleNotFoundError: No module named '_bz2
下载安装依赖
apt-get install libbz2-dev
进入python安装包下执行
./configure --with-bz2
make
sudo make install
2.ModuleNotFoundError: No module named '_lzma'
下载安装依赖
apt-get install liblzma-dev
pip install backports.lzma
修改lzma.py文件
vim /usr/local/lib/python3.10/lzma.py
将
from _lzma import *
from _lzma import _encode_filter_properties, _decode_filter_properties
改为
try:
from _lzma import *
from _lzma import _encode_filter_properties, _decode_filter_properties
except ImportError:
from backports.lzma import *
from backports.lzma import _encode_filter_properties, _decode_filter_properties
验证安装成功