文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

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

2023-09-03 15:26

关注

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。 

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

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

来源地址:https://blog.csdn.net/baidu_41774120/article/details/128588759

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯