PyTorch碎片:PyToch和Torchvision对应版本「建议收藏」

PyTorch碎片:PyToch和Torchvision对应版本「建议收藏」前言错误分析:安装pytorch或torchvision时,无法找到对应版本cuda可以找到,但是无法转为.cuda()以上两种或类似错误,一般由两个原因可供分析:cuda版本不合适,重新安装cuda和cudnnpytorch和torchvision版本没对应上pytorch和torchvision版本对应关系pytorchtorchvisionpythoncuda1.5.10.6.1>=3.69.2,10.1,10.21.5.0

大家好,又见面了,我是你们的朋友全栈君。

前言

  • 错误分析
  1. 安装pytorch或torchvision时,无法找到对应版本
  2. cuda可以找到,但是无法转为.cuda()
  • 以上两种或类似错误,一般由两个原因可供分析:
  1. cuda版本不合适,重新安装cuda和cudnn
  2. pytorch和torchvision版本没对应上

pytorch和torchvision版本对应关系

以下版本截止2022年5月5日

pytorch torchvision python cuda
1.11.0 0.10.0 >=3.6 11.3(不再支持CUDA10.2)
1.10.0,1.10.1 0.11.0 >=3.6 10.2,11.3
1.9.0 0.10.0 >=3.6 10.2,11.3
1.8.0 0.9.0 >=3.6 10.2,11.1
1.7.1 0.8.2 >=3.6 9.2, 10.1,10.2,11.0
1.7.0 0.8.0 >=3.6 9.2, 10.1,10.2,11.0
1.6.0 0.7.0 >=3.6 9.2, 10.1,10.2
1.5.1 0.6.1 >=3.6 9.2, 10.1,10.2
1.5.0 0.6.0 >=3.6 9.2, 10.1,10.2
1.4.0 0.5.0 ==2.7, >=3.5, <=3.8 9.2, 10.0
1.3.1 0.4.2 ==2.7, >=3.5, <=3.7 9.2, 10.0
1.3.0 0.4.1 ==2.7, >=3.5, <=3.7 9.2, 10.0
1.2.0 0.4.0 ==2.7, >=3.5, <=3.7 9.2, 10.0
1.1.0 0.3.0 ==2.7, >=3.5, <=3.7 9.0, 10.0
<1.0.1 0.2.2 ==2.7, >=3.5, <=3.7 9.0, 10.0

conda安装方法

根据pytorch官网方法安装,参考

https://pytorch.org/get-started

在这里插入图片描述

上面命令为安装最新的版本,为了能够应对各种不同硬件条件,常需要手动输入命令安装特定版本,可参考如下代码

# 安装pytorch==1.x.0 torchvision==0.x.0 cudatoolkit=10.x
conda install pytorch==1.x.0 torchvision==0.x.0 cudatoolkit=10.x -c pytorch

比如,举个更具体地例子:

# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch

后话

其实,环境中各版本对应关系出错,很容易导致各种乱七八糟地错误,所以尽量按照官方的要求安装。而且,新手还是建议使用次新版Pytorch,一方面强迫自己养成新coding习惯,一方面也为将来几年时间的代码不落伍提前准备。

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/152582.html原文链接:https://javaforall.net

(0)
上一篇 2022年6月24日 下午3:00
下一篇 2022年6月24日 下午3:16


相关推荐

  • 树莓派小车C语言循迹,自动循迹小车_单片机/STM32/树莓派/Arduino/开发板创意项目-聚丰项目-电子发烧友网…[通俗易懂]

    树莓派小车C语言循迹,自动循迹小车_单片机/STM32/树莓派/Arduino/开发板创意项目-聚丰项目-电子发烧友网…[通俗易懂]1金属探测模块采集与计算金属探测LDC1314是根据电磁感应原理制成的,将一金属置于变化的磁场当中时,根据电磁感应原理就会在金属内部产生涡流,涡流产生的磁场反过来又影响原磁场,这种变化可以转换为电压幅值的变化,供相关电路进行检测。通过改变金属和线圈之间的距离得到不同的值,对前端探测到的数据进行再处理和分析,当其中一个线圈探测到铁丝则让小车向相反方向前进,若发现附近有硬币存在该探测器发出声音警报。L…

    2022年5月6日
    65
  • 网页设计(新手入门)[通俗易懂]

    网页设计(新手入门)[通俗易懂]1.HTML的基本结构<!DOCTYPEHTML><html><head><title>百度一下,你就知道</title></head><body>……</body></html>HTML:是用于制作网页的超文本标记语…

    2022年4月19日
    45
  • c语言获取当前时间

    c语言获取当前时间首先介绍两个函数 1 intgettimeof structtimeva tv structtimezo tz 所属头文件 sys time h 其参数 tv 是保存获取时间结果的结构体 参数 tz 用于保存时区结果 structtimezo inttz minuteswest 格林威治时间往西方的时差 sys

    2026年3月19日
    2
  • 服务器中”系统平均负载 Load average“含义学习

    服务器中”系统平均负载 Load average“含义学习文章目录一、什么是系统平均负载二、衡量系统性能三、行车过桥(引用)四、自我总结一、什么是系统平均负载  uptime、w、top等命令都会有系统负载loadaverage的输出,系统平均负载被定义为在特定时间间隔内运行队列中的平均进程数,包括可运行状态和不可中断状态的平均进程数,也就是活跃进程数。它和cpu使用率没有直接的关系二、衡量系统性能  如果系统平均负载的数值除以CPU的数目高于…

    2025年11月7日
    6
  • python unicode编码转换utf-8编码_不成问题的问题人物解析

    python unicode编码转换utf-8编码_不成问题的问题人物解析Unicode也叫万国码、单一码,是计算机科学领域里的一项业界标准,包括字符集、编码方案等。对于世界上所有的语言文字再unicode中都可以查看到。【汉】字的编码解释官网https://www.unicode.org/cgi-bin/GetUnihanData.pl?codepoint=6C49unicode编码就是为了统一世界上的编码,有一个统一的规范。但是它还存在一些问题。Unicode的问题需要注意的是,Unicode只是一个符号集,它只规定了符号的二进制代码,却没有规定这个二进制代码应该如何存

    2022年9月30日
    6
  • 刚刚弄的51cto 谢谢我的兄弟

    刚刚弄的51cto 谢谢我的兄弟

    2021年8月11日
    66

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号