热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

Python安装torch(含torch和torchvision对应版本)

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

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


torch torchvision python
master / nightlymaster / nightly>=3.6
1.7.1 0.8.2>=3.6
1.7.0 0.8.1>=3.6
1.7.00.8.0>=3.6
1.6.00.7.0>=3.6
1.5.10.6.1>=3.5
1.5.00.6.0>=3.5
1.4.00.5.0  ==2.7, >=3.5, <=3.8
1.3.1 0.4.2==2.7, >=3.5, <=3.7
1.3.00.4.1==2.7, >=3.5, <=3.7
1.2.00.4.0==2.7, >=3.5, <=3.7
1.1.0 0.3.0==2.7, >=3.5, <=3.7
<=1.0.10.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) pythonPython 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。 

 


推荐阅读
author-avatar
手机用户2502937657
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有