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 版本方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python 正则表达式操作指南
May 04 Python
解决Django migrate No changes detected 不能创建表的问题
May 27 Python
浅谈Django的缓存机制
Aug 23 Python
详解Python做一个名片管理系统
Mar 14 Python
Python 中的参数传递、返回值、浅拷贝、深拷贝
Jun 25 Python
简单了解django orm中介模型
Jul 30 Python
python中struct模块之字节型数据的处理方法
Aug 27 Python
Python 读取 YUV(NV12) 视频文件实例
Dec 09 Python
Python实现剪刀石头布小游戏(与电脑对战)
Dec 31 Python
django之导入并执行自定义的函数模块图解
Apr 01 Python
如何用Python 实现全连接神经网络(Multi-layer Perceptron)
Oct 15 Python
Python实现Appium端口检测与释放的实现
Dec 31 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
You might like
PHP得到某段时间区间的时间戳 php定时任务
2012/04/12 PHP
php遍历目录与文件夹的多种方法详解
2013/11/14 PHP
图文介绍PHP添加Redis模块及连接
2015/07/28 PHP
Yii2中事务的使用实例代码详解
2016/09/07 PHP
php根据数据id自动生成编号的实现方法
2016/10/16 PHP
javascript下string.format函数补充
2010/08/24 Javascript
如何让浏览器支持jquery ajax load 前进、后退功能
2014/06/12 Javascript
基于jQuery1.9版本如何判断浏览器版本类型
2016/01/12 Javascript
zTree插件下拉树使用入门教程
2016/04/11 Javascript
浅谈在js传递参数中含加号(+)的处理方式
2016/10/11 Javascript
基于原生js淡入淡出函数封装(兼容IE)
2016/10/20 Javascript
JS中微信小程序自定义底部弹出框
2016/12/22 Javascript
bootstrap下拉菜单使用方法解析
2017/01/13 Javascript
微信小程序 出现47001 data format error原因解决办法
2017/03/10 Javascript
jQuery进阶实践之利用最优雅的方式如何写ajax请求
2017/12/20 jQuery
如何写好一个vue组件,老夫的一年经验全在这了(推荐)
2019/05/18 Javascript
layui数据表格实现重载数据表格功能(搜索功能)
2019/07/27 Javascript
简单了解Vue computed属性及watch区别
2020/07/10 Javascript
Vue表单提交点击事件只允许点击一次的实例
2020/10/23 Javascript
[51:05]DOTA2上海特级锦标赛主赛事日 - 5 败者组决赛Liquid VS EG第一局
2016/03/06 DOTA
python黑魔法之参数传递
2016/02/12 Python
PyQt5每天必学之关闭窗口
2018/04/19 Python
PyQt5+requests实现车票查询工具
2019/01/21 Python
python使用Plotly绘图工具绘制气泡图
2019/04/01 Python
用Python实现二叉树、二叉树非递归遍历及绘制的例子
2019/08/09 Python
基于keras中的回调函数用法说明
2020/06/17 Python
学生自我鉴定范文
2013/10/04 职场文书
学习党章思想汇报
2014/01/07 职场文书
文明礼仪伴我行演讲稿
2014/05/12 职场文书
党建工作整改措施
2014/10/28 职场文书
公务员年度考核评语
2014/12/31 职场文书
2015年副班长工作总结
2015/05/15 职场文书
人民的好儿女观后感
2015/06/18 职场文书
2016年感恩节活动总结大全
2016/04/01 职场文书
CSS3实现的水平标题菜单
2021/04/14 HTML / CSS
Flask搭建一个API服务器的步骤
2021/05/28 Python