首页
所有分类
TAGS
文字工具 EMOJI BIBLE
编程 Python

pytorch 查看cuda 版本方式

Posted in Python onJune 23, 2020

由于pytorch的whl 安装包名字都一样，所以我们很难区分到底是基于cuda 的哪个版本。

有一条指令可以查看

import torch
print(torch.version.cuda)

补充知识：pytorch：网络定义参数的时候后面不能加".cuda()"

pytorch定义网络__init__()的时候，参数不能加“cuda()", 不然参数不包含在state_dict()中，比如下面这种写法是错误的

self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True).cuda()

应该去掉".cuda()"

self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True)

以上这篇pytorch 查看cuda 版本方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

pytorch 查看cuda 版本方式

- Author -

峰峰晶晶

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐
Python中的特殊语法：filter、map、reduce、lambda介绍
Apr 14 Python
python动态性强类型用法实例
May 09 Python
在Django中创建动态视图的教程
Jul 15 Python
Python列表解析配合if else的方法
Jun 23 Python
python中的tcp示例详解
Dec 09 Python
python3.5安装python3-tk详解
Apr 26 Python
python 含子图的gif生成时内存溢出的方法
Jul 07 Python
利用 Flask 动态展示 Pyecharts 图表数据方法小结
Sep 04 Python
Python hashlib加密模块常用方法解析
Dec 18 Python
Pandas读取csv时如何设置列名
Jun 02 Python
Python+logging输出到屏幕将log日志写入文件
Nov 11 Python
发工资啦！教你用Python实现邮箱自动群发工资条
May 10 Python
Spring @Enable模块驱动原理及使用实例
Jun 23 #Python
PyTorch-GPU加速实例
Jun 23 #Python
Python基于yaml文件配置logging日志过程解析
Jun 23 #Python
解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题
Jun 23 #Python
Python 程序报错崩溃后如何倒回到崩溃的位置(推荐)
Jun 23 #Python
浅谈pytorch中的BN层的注意事项
Jun 23 #Python
Python3与fastdfs分布式文件系统如何实现交互
Jun 23 #Python
mounted(1) Mybatis-Plus(3) set(1) 脚手架(2) Next.js(1) frozenset(1) create(1) TableField(1) SingleFlight(2) el-table(1)
You might like
php中理解print EOT分界符和echo EOT的用法区别小结
2010/02/21 PHP
PHP下获取上个月、下个月、本月的日期(strtotime,date)
2014/02/02 PHP
PHP中substr()与explode()函数用法分析
2014/11/24 PHP
php计算一个文件大小的方法
2015/03/30 PHP
PHP读书笔记_运算符详解
2016/07/01 PHP
Thinkphp极验滑动验证码实现步骤解析
2020/11/24 PHP
用js判断用户浏览器是否是XP SP2的IE6
2007/03/08 Javascript
jQuery html() in Firefox (uses .innerHTML) ignores DOM changes
2010/03/05 Javascript
Chrome中JSON.parse的特殊实现
2011/01/12 Javascript
JavaScript中的全局对象介绍
2015/01/01 Javascript
jscript读写二进制文件的方法
2015/04/22 Javascript
jQuery基于cookie实现的购物车实例分析
2015/12/24 Javascript
星期几的不同脚本写法(推荐)
2016/06/01 Javascript
canvas仿iwatch时钟效果
2017/03/06 Javascript
深究AngularJS之ui-router详解
2017/06/13 Javascript
ES6学习教程之模板字符串详解
2017/10/09 Javascript
详解http访问解析流程原理
2017/10/18 Javascript
详解如何在react中搭建d3力导向图
2018/01/12 Javascript
layerUI下的绑定事件实例代码
2018/08/17 Javascript
微信小程序实现弹出菜单动画
2019/06/21 Javascript
jQuery HTML设置内容和属性操作实例分析
2020/05/20 jQuery
解决vue-loader加载不上的问题
2020/10/21 Javascript
python实现数通设备tftp备份配置文件示例
2014/04/02 Python
Django中使用locals()函数的技巧
2015/07/16 Python
解决Python2.7读写文件中的中文乱码问题
2018/04/12 Python
python 分离文件名和路径以及分离文件名和后缀的方法
2018/10/21 Python
python之信息加密题目详解
2019/06/26 Python
python射线法判断检测点是否位于区域外接矩形内
2019/06/28 Python
keras 模型参数,模型保存,中间结果输出操作
2020/07/06 Python
Python暴力破解Mysql数据的示例
2020/11/09 Python
python爬取抖音视频的实例分析
2021/01/19 Python
html5 web本地存储将取代我们的cookie
2012/12/26 HTML / CSS
乔丹诺(Giordano)酒庄德国官网：找到最好的意大利葡萄酒
2017/12/28 全球购物
.NET是怎么支持多种语言的
2015/02/24 面试题
领导干部整治奢华浪费之风思想汇报
2014/10/07 职场文书
医院领导班子四风问题对照检查材料
2014/10/26 职场文书