pytorch和torchvision安装

80 阅读 0 评论 53 点赞

我是靠谱客的博主贤惠天空，这篇文章主要介绍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/

以上就是贤惠天空最近收集整理的关于pytorch和torchvision安装的全部内容，更多相关pytorch和torchvision安装内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。