ubuntu安装深度学习框架torch

---

前言

ubuntu安装torch，首先先从官网查看对应的版本

---

一、最新版安装

sudo pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

本人安装版本，针对电脑安装

python3.8以下
sudo pip3 install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
python3.8以上
sudo pip3 install torch==1.4.0 torchvision==0.5.0

报错信息

1、 安装torch1.9.0和torchvision0.10.0时会报错 ModuleNotFoundError: No module named ‘_lzma’
解决方案：

# centos系统执行
yum install xz-devel -y
yum install python-backports-lzma -y
pip install backports.lzma

# ubuntu系统执行
apt-get install liblzma-dev -y
pip install backports.lzma

gedit /usr/local/lib/python3.8/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