编程 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程序的执行效率的教程

Apr 02 Python

Python实现把json格式转换成文本或sql文件

Jul 10 Python

Python 40行代码实现人脸识别功能

Apr 02 Python

深入理解Python爬虫代理池服务

Feb 28 Python

详解PyCharm安装MicroPython插件的教程

Jun 24 Python

python实现比较类的两个instance(对象)是否相等的方法分析

Jun 26 Python

Python3实现汉语转换为汉语拼音

Jul 08 Python

Python时间序列缺失值的处理方法（日期缺失填充）

Aug 11 Python

numpy:np.newaxis 实现将行向量转换成列向量

Nov 30 Python

如何使用python传入不确定个数参数

Feb 18 Python

PyQt 如何创建自定义QWidget

Mar 24 Python

教你利用Selenium+python自动化来解决pip使用异常

May 20 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

理解PHP5中static和const关键字的区别

2007/03/19 PHP

PHP 删除一个目录及目录下的所有文件的函数代码

2010/05/26 PHP

PHP中对用户身份认证实现两种方法

2011/06/04 PHP

如何用php生成扭曲及旋转的验证码图片

2013/06/07 PHP

php 无限级分类，超级简单的无限级分类，支持输出树状图

2014/06/29 PHP

PHP5中使用mysqli的prepare操作数据库的介绍

2019/03/18 PHP

深入理解Javascript作用域与变量提升

2013/12/09 Javascript

jquery+CSS3实现淘宝移动网页菜单效果

2015/08/31 Javascript

jQuery中inArray方法注意事项分析

2016/01/25 Javascript

JavaScript解八皇后问题的方法总结

2016/06/12 Javascript

把普通对象转换成json格式的对象的简单实例

2016/07/04 Javascript

VC调用javascript的几种方法(推荐)

2016/08/09 Javascript

浅谈JavaScript的闭包函数

2016/12/08 Javascript

AngularJS的ng-repeat指令与scope继承关系实例详解

2017/01/21 Javascript

Canvas实现动态的雪花效果

2017/02/13 Javascript

微信小程序实现列表项滑动显示删除按钮的功能

2017/04/13 Javascript

利用vue开发一个所谓的数独方法实例

2017/12/21 Javascript

简述vue中的config配置

2018/01/23 Javascript

详解react关于事件绑定this的四种方式

2018/03/09 Javascript

JS中min函数实例讲解

2019/02/18 Javascript

JavaScript 防盗链的原理以及破解方法

2020/12/29 Javascript

python中根据字符串调用函数的实现方法

2016/06/12 Python

Flask框架URL管理操作示例【基于@app.route】

2018/07/23 Python

Python使用uuid库生成唯一标识ID

2020/02/12 Python

在Windows上安装和配置 Jupyter Lab 作为桌面级应用程序教程

2020/04/22 Python

Python3如何使用tabulate打印数据

2020/09/25 Python

美国鲍勃商店：Bob’s Stores

2018/07/22 全球购物

Farfetch巴西官网：奢侈品牌时尚购物平台

2020/10/19 全球购物

linux系统都有哪些运行级别

2016/03/26 面试题

介绍下Lucene建立索引的过程

2016/03/02 面试题

本科生导师推荐信范文

2014/05/18 职场文书

民主生活会对照检查材料思想汇报

2014/09/27 职场文书

小学家长通知书评语

2014/12/31 职场文书

合作与交流自我评价

2015/03/09 职场文书

2015年检验科工作总结

2015/04/27 职场文书

Pycharm连接远程服务器并远程调试的全过程

2021/06/24 Python