编程 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 03 Python

在Django框架中伪造捕捉到的URLconf值的方法

Jul 18 Python

详解Python3中字符串中的数字提取方法

Jan 14 Python

linux平台使用Python制作BT种子并获取BT种子信息的方法

Jan 20 Python

浅析Python中的赋值和深浅拷贝

Aug 15 Python

rabbitmq(中间消息代理)在python中的使用详解

Dec 14 Python

Python OpenCV利用笔记本摄像头实现人脸检测

Aug 20 Python

python操作docx写入内容,并控制文本的字体颜色

Feb 13 Python

python学生管理系统的实现

Apr 05 Python

python程序输出无内容的解决方式

Apr 09 Python

Python Flask异步发送邮件实现方法解析

Aug 01 Python

Python&Matlab实现灰狼优化算法的示例代码

Mar 21 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基本语法实例总结

2016/09/09 PHP

PHP实现验证码校验功能

2017/11/16 PHP

Laravel框架实现超简单的分页效果示例

2019/02/08 PHP

php设计模式之原型模式分析【星际争霸游戏案例】

2020/03/23 PHP

PHP dirname(__FILE__)原理及用法解析

2020/10/28 PHP

写的htc的数据表格

2007/01/20 Javascript

自适应高度框架 ----属个人收藏内容

2007/01/22 Javascript

JavaScript Undefined，Null类型和NaN值区别

2008/10/22 Javascript

图片在浏览器中底部对齐解决方法之一

2011/11/30 Javascript

jQuery在iframe中无法弹出对话框的解决方法

2014/01/12 Javascript

JS 打印功能代码可实现打印预览、打印设置等

2014/10/31 Javascript

JS实现的打字机效果完整实例

2016/06/20 Javascript

bootstrap是什么_动力节点Java学院整理

2017/07/14 Javascript

基于javaScript的this指向总结

2017/07/22 Javascript

vue+vux实现移动端文件上传样式

2017/07/28 Javascript

jquery easyui如何实现格式化列

2017/07/30 jQuery

webpack实现热加载自动刷新的方法

2017/07/30 Javascript

10种JavaScript最常见的错误(小结)

2019/06/21 Javascript

对TypeScript库进行单元测试的方法

2019/07/18 Javascript

TypeScript的安装、使用、自动编译的实现

2020/04/10 Javascript

[37:21]完美世界DOTA2联赛PWL S2 Inki vs Magma 第二场 11.22

2020/11/24 DOTA

Python基于scrapy采集数据时使用代理服务器的方法

2015/04/16 Python

[原创]Python入门教程5. 字典基本操作【定义、运算、常用函数】

2018/11/01 Python

详解python中的生成器、迭代器、闭包、装饰器

2019/08/22 Python

Python 网络编程之UDP发送接收数据功能示例【基于socket套接字】

2019/10/11 Python

使用python 将图片复制到系统剪贴中

2019/12/13 Python

Python守护进程实现过程详解

2020/02/10 Python

Python3安装模块报错Microsoft Visual C++ 14.0 is required的解决方法

2020/07/28 Python

python speech模块的使用方法

2020/09/09 Python

DeinDesign德国：设计自己的手机壳

2019/12/14 全球购物

打架检讨书50字

2014/01/11 职场文书

护士毕业实习感言

2014/03/05 职场文书

事业单位个人查摆问题及整改措施

2014/10/28 职场文书

幼儿园小班个人总结

2015/02/12 职场文书

幼儿园教学反思范文

2016/03/02 职场文书

OpenCV-Python使用cv2实现傅里叶变换

2021/06/09 Python