pytorch和torchvision安装

目录

1. pytorch和torchvision简介       

2. pytorch和torchvision安装

---

1. pytorch和torchvision简介       

Pytorch是一个开源的python机器学习库，具有强大的GPU加速的张量计算，而torchvision是pytorch中专门用来处理图像的库，主要包含四大类：

torchvision.datasets: 主要用来进行数据加载

torchvision.model: 主要用来进行模型加载

torchvision.transforms: 主要用于图像相关的操作

torchvision.utils: 其他模块操作

2. pytorch和torchvision安装

使用pip安装pytorch和torchvision,如未安装pip工具，则需先安装pip，安装命令：

sudo apt install python-pip

国内由于网速的限制，使用命令安装时需在后面加上清华的镜像，这样下载速度会大大提升，安装更容易成功。

pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/

此命令在安装torchvision的同时，会把torch一起安装。安装结束后重新开启一个终端并进入python环境，输入以下命令，如果没报异常，则说明安装成功了。

import torch
import torchvision

查看pytorch版本号

import torch
print(torch.__version__)

由于python2已不再更新，为避免兼容性问题，建议在安装时将pytorch安装在python3下，使用python3安装pytorch.

sudo apt install python3-pip

pip3 install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/