编程 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实现批量监控网站

Sep 09 Python

Python在图片中添加文字的两种方法

Apr 29 Python

python实现简易版计算器

Jun 22 Python

Python发送http请求解析返回json的实例

Mar 26 Python

选择Python写网络爬虫的优势和理由

Jul 07 Python

TensorFlow车牌识别完整版代码（含车牌数据集）

Aug 05 Python

python django生成迁移文件的实例

Aug 31 Python

opencv3/Python 稠密光流calcOpticalFlowFarneback详解

Dec 11 Python

使用Tensorboard工具查看Loss损失率

Feb 15 Python

MATLAB数学建模之画图汇总

Jul 16 Python

Python+Selenium随机生成手机验证码并检查页面上是否弹出重复手机号码提示框

Sep 21 Python

python保存图片的四个常用方法

Feb 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可逆加密/解密函数分享

2012/09/25 PHP

php 模拟POST提交的2种方法详解

2013/06/17 PHP

注意:php5.4删除了session_unregister函数

2013/08/05 PHP

PHP URL路由类实例

2013/11/12 PHP

ThinkPHP自动转义存储富文本编辑器内容导致读取出错的解决方法

2014/08/08 PHP

php中异常处理方法小结

2015/01/09 PHP

js如何获取file控件的完整路径具体实现代码

2013/05/15 Javascript

jQuery动画效果-fadeIn fadeOut淡入浅出示例代码

2013/08/28 Javascript

jquery ready函数、css函数及text()使用示例

2013/09/27 Javascript

javascript 用函数语句和表达式定义函数的区别详解

2014/01/06 Javascript

jQuery实现设置、移除文本框默认值功能

2015/01/13 Javascript

JS简单编号生成器实现方法(附demo源码下载)

2016/04/05 Javascript

封装获取dom元素的简单实例

2016/07/08 Javascript

KnockoutJS 3.X API 第四章之数据控制流with绑定

2016/10/10 Javascript

JS实现图片高斯模糊切换效果的焦点图实例

2017/01/21 Javascript

vuejs响应用户事件（如点击事件）

2017/03/14 Javascript

js轮播图透明度切换（带上下页和底部圆点切换）

2017/04/27 Javascript

jQuery动态添加.active 实现导航效果代码思路详解

2017/08/29 jQuery

全面介绍vue 全家桶和项目实例

2017/12/27 Javascript

关于Angularjs中跨域设置白名单问题

2018/04/17 Javascript

解决layui表格内文本超出隐藏的问题

2019/09/12 Javascript

layer ui 导入文件之前传入数据的实例

2019/09/23 Javascript

在react-antd中弹出层form内容传递给父组件的操作

2020/10/24 Javascript

[06:45]2018DOTA2亚洲邀请赛 4.5 SOLO赛 Sccc vs Maybe

2018/04/06 DOTA

Python高级应用实例对比：高效计算大文件中的最长行的长度

2014/06/08 Python

python在windows下实现ping操作并接收返回信息的方法

2015/03/20 Python

python的Crypto模块实现AES加密实例代码

2018/01/22 Python

python实现感知机模型的示例

2020/09/30 Python

用python查找统一局域网下ip对应的mac地址

2021/01/13 Python

荷兰最大的多品牌男装连锁店：Adam Brandstore

2019/12/31 全球购物

如何在Shell脚本中使用函数

2015/09/06 面试题

校园文化建设方案

2014/02/03 职场文书

压缩Redis里的字符串大对象操作

2021/06/23 Redis

vue ref如何获取子组件属性值

2022/03/31 Vue.js

阿里云ECS云服务器快照的概念以及如何使用

2022/04/21 Servers

Python使用openpyxl模块处理Excel文件

2022/06/05 Python