编程 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 解析XML文件

Apr 15 Python

Python中对列表排序实例

Jan 04 Python

Django中使用locals()函数的技巧

Jul 16 Python

Python实现对特定列表进行从小到大排序操作示例

Feb 11 Python

用python打印菱形的实操方法和代码

Jun 25 Python

python 自动轨迹绘制的实例代码

Jul 05 Python

用django设置session过期时间的方法解析

Aug 05 Python

python中matplotlib条件背景颜色的实现

Sep 02 Python

Python time库基本使用方法分析

Dec 13 Python

详解使用python3.7配置开发钉钉群自定义机器人(2020年新版攻略)

Apr 01 Python

Python多线程thread及模块使用实例

Apr 28 Python

浅谈django channels 路由误导

May 28 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 默默经典版本

2009/08/04 PHP

php设计模式 Command(命令模式)

2011/06/17 PHP

浅谈php中mysql与mysqli的区别分析

2013/06/10 PHP

PHP实现基于3DES算法加密解密字符串示例

2018/08/24 PHP

php进程(线程)通信基础之System V共享内存简单实例分析

2019/11/09 PHP

在Javascript中定义对象类别

2006/12/22 Javascript

javascript 装载iframe子页面，自适应高度

2009/03/20 Javascript

几个javascript操作word的参考代码

2009/10/26 Javascript

jquery $(document).ready() 与window.onload的区别

2009/12/28 Javascript

JavaScript异步调用定时方法并停止该方法实现代码

2012/03/16 Javascript

JS二维数组的定义说明

2014/03/03 Javascript

JS实现合并两个数组并去除重复项只留一个的方法

2015/12/17 Javascript

Node.js connect ECONNREFUSED错误解决办法

2016/09/15 Javascript

jQuery ui autocomplete选择列表被Bootstrap模态窗遮挡的完美解决方法

2016/09/23 Javascript

js实现碰撞检测特效代码分享

2016/10/16 Javascript

nodejs和php实现图片访问实时处理

2017/01/05 NodeJs

微信小程序文章详情页面实现代码

2018/09/10 Javascript

微信小程序实现打卡日历功能

2020/09/21 Javascript

Vue触发式全局组件构建的方法

2018/11/28 Javascript

详解vue后台系统登录态管理

2019/04/02 Javascript

使用vue编写h5公众号跳转小程序的实现代码

2020/11/27 Vue.js

vue中实现点击空白区域关闭弹窗的两种方法

2020/12/30 Vue.js

Python基于Socket实现的简单聊天程序示例

2017/08/05 Python

python3下使用cv2.imwrite存储带有中文路径图片的方法

2018/05/10 Python

python操作小程序云数据库实现简单的增删改查功能

2019/06/06 Python

python之当你发现QTimer不能用时的解决方法

2019/06/21 Python

Python中利用LSTM模型进行时间序列预测分析的实现

2019/07/26 Python

Pytorch训练过程出现nan的解决方式

2020/01/02 Python

Ubuntu中配置TensorFlow使用环境的方法

2020/04/21 Python

Django Form设置文本框为readonly操作

2020/07/03 Python

HTML5 placeholder(空白提示)属性介绍

2013/08/07 HTML / CSS

Opodo意大利：欧洲市场上领先的在线旅行社

2019/10/24 全球购物

WEB控件及HTML服务端控件能否调用客户端方法？如果能，请解释如何调用？

2015/08/25 面试题

党员入党表决心的话

2014/03/11 职场文书

python 模块重载的五种方法

2021/04/24 Python

JavaScript声明变量和数据类型的转换

2022/04/12 Javascript