编程 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写的服务监控程序实例

Jan 31 Python

Python整型运算之布尔型、标准整型、长整型操作示例

Jul 21 Python

详解Python函数可变参数定义及其参数传递方式

Aug 02 Python

对TensorFlow的assign赋值用法详解

Jul 30 Python

PyQt5实现简单数据标注工具

Mar 18 Python

Python Excel处理库openpyxl使用详解

May 09 Python

python opencv 批量改变图片的尺寸大小的方法

Jun 28 Python

pytorch numpy list类型之间的相互转换实例

Aug 18 Python

Python单例模式的四种创建方式实例解析

Mar 04 Python

python实现简单遗传算法

Sep 18 Python

Python类的继承super相关原理解析

Oct 22 Python

用Python制作音乐海报

Jan 26 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

基于OpenCV的PHP图像人脸识别技术

2009/10/11 PHP

php XPath对XML文件查找及修改实现代码

2011/07/27 PHP

PHP __autoload()方法真的影响性能吗?

2012/03/30 PHP

AJAX的跨域访问-两种有效的解决方法介绍

2013/06/22 PHP

PHP开发实现微信退款功能示例

2017/11/25 PHP

Laravel框架源码解析之反射的使用详解

2020/05/14 PHP

判断页面是关闭还是刷新的js代码

2007/01/28 Javascript

syntaxhighlighter 使用方法

2007/07/02 Javascript

JavaScript入门教程(2) JS基础知识

2009/01/31 Javascript

网页和浏览器兼容性问题汇总(draft1)

2009/06/01 Javascript

[JSF]使用DataModel处理表行事件的实例代码

2013/08/05 Javascript

12种不宜使用的Javascript语法整理

2013/11/04 Javascript

基于jquery实现轮播焦点图插件

2016/03/31 Javascript

浅谈JavaScript中数组的增删改查

2016/06/20 Javascript

AngularJS基础 ng-keypress 指令简单示例

2016/08/02 Javascript

轻松掌握JavaScript装饰者模式

2016/08/27 Javascript

用js实现博客打赏功能

2016/10/24 Javascript

fullCalendar中文API官方文档

2017/02/07 Javascript

vue axios 二次封装的示例代码

2017/12/08 Javascript

JavaScript DOM元素常见操作详解【添加、删除、修改等】

2018/05/09 Javascript

Angular7.2.7路由使用初体验

2019/03/01 Javascript

javascript获取元素的计算样式

2019/05/24 Javascript

Vue项目总结之webpack常规打包优化方案

2019/06/06 Javascript

JavaScript Array对象基本方法详解

2019/09/03 Javascript

js动态添加带圆圈序号列表的实例代码

2021/02/18 Javascript

Python version 2.7 required, which was not found in the registry

2014/08/26 Python

Python3中类、模块、错误与异常、文件的简易教程

2017/11/20 Python

Python Cookie 读取和保存方法

2018/12/28 Python

详解从Django Allauth中进行登录改造小结

2019/12/18 Python

解决IDEA 的 plugins 搜不到任何的插件问题

2020/05/04 Python

广州某公司软件工程师面试题

2014/12/22 面试题

拖鞋店创业计划书

2014/01/15 职场文书

部队反四风对照检查材料

2014/09/26 职场文书

党员群众路线个人整改措施思想汇报

2014/10/12 职场文书

采购部年度工作总结

2015/08/13 职场文书

从结婚开始的恋爱故事。小说《我的美好婚事》TV动画化决定

2022/04/07 日漫