Python安装torch（含torch和torchvision对应版本）

文章Python安装torch（含torch和torchvision对应版本）分享给大家，欢迎收藏Python资料网，专注分享技术知识

torch 、torchvision和python的对应版本如下：

torch	torchvision	python
master / nightly	master / nightly	>=3.6
1.7.1	0.8.2	>=3.6
1.7.0	0.8.1	>=3.6
1.7.0	0.8.0	>=3.6
1.6.0	0.7.0	>=3.6
1.5.1	0.6.1	>=3.5
1.5.0	0.6.0	>=3.5
1.4.0	0.5.0	==2.7, >=3.5, <=3.8
1.3.1	0.4.2	==2.7, >=3.5, <=3.7
1.3.0	0.4.1	==2.7, >=3.5, <=3.7
1.2.0	0.4.0	==2.7, >=3.5, <=3.7
1.1.0	0.3.0	==2.7, >=3.5, <=3.7
<=1.0.1	0.2.2	==2.7, >=3.5, <=3.7

方法一（通过下载到本地安装）：

torch 和 torchvision 的下载网址：

https://download.pytorch.org/whl/torch_stable.html

这里以 torch1.2.0 和 torchvision0.4.0 的安装（python3.7）为例： 

下载自己对应版本即可。

注：cu100说明是cuda10.0版本，cu90是cuda9.0版本，37说明适配python3.7版本，35就是适配python3.5版本。

 下载好之后，打开Anaconda中创建的虚拟环境，然后通过cd /d D:...（以D盘为例）进入到你下载的位置，之后pip安装即可。

通过下面的代码可以查看安装好的torch版本：

conda activate pytorch       # 激活环境

(pytorch) python

Python 3.7.0 

>>> import torch
>>> print(torch.__version__)
1.2.0
# 显示安装torch版本

>>> print(torch.cuda.is_available())
True
# GPU能使用显示ture

>>> exit(0)
# 退出编写

 出现版本号说明安装成功了，就可以使用了。

方法二（直接通过conda命令安装）：

pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/

在上面的命令中输入需要安装的版本号，等待安装就可以了。

注：需要加上后面的清华镜像，不然会出现安装不成功的问题。 

附上一张图，这里我安装的是torch1.4.0和torchvision0.5.0，环境是python3.8。 

如果使用conda直接安装还是有问题，可以看看下面这篇文章：

(11条消息) 保姆级别的Anaconda/pip高速安装pytorch1.7.1_Shlily.的博客-CSDN博客_pip torch 1.7.1 

版权声明：本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若内容造成侵权/违法违规/事实不符，请联系邮箱：jacktools123@163.com进行投诉反馈，一经查实，立即删除！