pytorch和torchvision安装

98 阅读 0 评论 65 点赞

我是靠谱客的博主贤惠天空，最近开发中收集的这篇文章主要介绍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安装所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

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