编程 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 相关文章推荐

忘记ftp密码使用python ftplib库暴力破解密码的方法示例

Jan 22 Python

Python实现方便使用的级联进度信息实例

May 05 Python

python版微信跳一跳游戏辅助

Jan 11 Python

python中的句柄操作的方法示例

Jun 20 Python

Python使用pyserial进行串口通信的实例

Jul 02 Python

爬虫代理池Python3WebSpider源代码测试过程解析

Dec 20 Python

Tensorflow轻松实现XOR运算的方式

Feb 03 Python

在python中使用pymysql往mysql数据库中插入(insert)数据实例

Mar 02 Python

python安装dlib库报错问题及解决方法

Mar 16 Python

pyecharts动态轨迹图的实现示例

Apr 17 Python

python链表类中获取元素实例方法

Feb 23 Python

Python中使用tkFileDialog实现文件选择、保存和路径选择

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

特转载一高手总结PHP学习资源和链接.

2006/12/05 PHP

一个数据采集类

2007/02/14 PHP

探讨如何在PHP开启gzip页面压缩实例

2013/06/09 PHP

PHP贪婪算法解决0-1背包问题实例分析

2015/03/23 PHP

深入理解PHP内核(一)

2015/11/10 PHP

php array_walk_recursive 使用自定的函数处理数组中的每一个元素

2016/11/16 PHP

PHP判断是手机端还是PC端 PHP判断是否是微信浏览器

2017/03/15 PHP

js滚动条多种样式,推荐

2007/02/05 Javascript

jquery 防止表单重复提交代码

2010/01/21 Javascript

javascript用户注册提示效果的简单实例

2013/08/17 Javascript

解决jQuery动态获取手机屏幕高和宽的问题

2014/05/07 Javascript

JS实现带关闭功能的阿里妈妈网站顶部滑出banner工具条代码

2015/09/17 Javascript

一道常被人轻视的web前端常见面试题(JS)

2016/02/15 Javascript

有关JavaScript中call()和apply() 的一些理解

2016/05/20 Javascript

jQuery获取file控件中图片的宽高与大小

2016/08/04 Javascript

Angular的$http的ajax的请求操作（推荐）

2017/01/10 Javascript

Material(包括Material Icon)在Angular2中的使用详解

2018/02/11 Javascript

微信公众号平台接口开发菜单管理的实现

2019/08/14 Javascript

通过高德地图API获得某条道路上的所有坐标用于描绘道路的方法

2020/08/24 Javascript

详解JavaScript中的数据类型，以及检测数据类型的方法

2020/09/17 Javascript

[53:15]2018DOTA2亚洲邀请赛3月29日小组赛A组 KG VS OG

2018/03/30 DOTA

Python利用pyHook实现监听用户鼠标与键盘事件

2014/08/21 Python

Python实现Linux下守护进程的编写方法

2014/08/22 Python

Python素数检测的方法

2015/05/11 Python

Python使用回溯法子集树模板获取最长公共子序列（LCS）的方法

2017/09/08 Python

Python使用scrapy爬取阳光热线问政平台过程解析

2019/08/14 Python

PyCharm2019安装教程及其使用(图文教程)

2019/09/29 Python

python实现scrapy爬虫每天定时抓取数据的示例代码

2021/01/27 Python

pycharm 使用anaconda为默认环境的操作

2021/02/05 Python

意大利一家专营包包和配饰的网上商店：Borse Last Minute

2019/08/26 全球购物

Myprotein中国网站：欧洲畅销运动营养品牌

2021/02/11 全球购物

八一建军节部队活动方案

2014/02/04 职场文书

学习全国两会精神心得体会范文

2014/03/17 职场文书

大专学生求职自荐信

2014/07/06 职场文书

中学生清明节演讲稿

2015/03/18 职场文书

民间借贷借条如何写

2015/05/26 职场文书