编程 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爬虫之抓取百度贴吧代码分享

Nov 06 Python

Python的Bottle框架中获取制定cookie的教程

Apr 24 Python

Python通过正则表达式选取callback的方法

Jul 18 Python

Python编码类型转换方法详解

Jul 01 Python

Python判断某个用户对某个文件的权限

Oct 13 Python

手把手教你python实现SVM算法

Dec 27 Python

numpy中实现二维数组按照某列、某行排序的方法

Apr 04 Python

Python使用requests提交HTTP表单的方法

Dec 26 Python

python basemap 画出经纬度并标定的实例

Jul 09 Python

Python Django实现layui风格+django分页功能的例子

Aug 29 Python

pygame实现贪吃蛇游戏（上）

Oct 29 Python

Python编译成.so文件进行加密后调用的实现

Dec 23 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代码

2010/02/16 PHP

php操作SVN版本服务器类代码

2011/11/27 PHP

回帖脱衣服的图片实现代码

2014/02/15 PHP

Linux系统下PHP-FPM的安装和配置教程

2015/08/17 PHP

JS模拟多线程

2007/02/07 Javascript

js新闻滚动 js如何实现新闻滚动效果

2013/01/07 Javascript

javascript实现yield的方法

2013/11/06 Javascript

node.js中的fs.chownSync方法使用说明

2014/12/16 Javascript

Jquery插件实现点击获取验证码后60秒内禁止重新获取

2015/03/13 Javascript

js实现使用鼠标拖拽切换图片的方法

2015/05/04 Javascript

vue中tab选项卡的实现思路

2018/11/25 Javascript

JS数组及对象遍历方法代码汇总

2020/06/16 Javascript

[54:30]Liquid vs Newbee 2019国际邀请赛小组赛 BO2 第二场 8.15

2019/08/16 DOTA

在树莓派2或树莓派B+上安装Python和OpenCV的教程

2015/03/30 Python

python排序方法实例分析

2015/04/30 Python

Python的Flask框架标配模板引擎Jinja2的使用教程

2016/07/12 Python

用Python解决计数原理问题的方法

2016/08/04 Python

基于Python实现对PDF文件的OCR识别

2016/08/05 Python

解决Python中定时任务线程无法自动退出的问题

2019/02/18 Python

Python实现使用request模块下载图片demo示例

2019/05/24 Python

pandas DataFrame的修改方法(值、列、索引)

2019/08/02 Python

Django中自定义admin Xadmin的实现代码

2019/08/09 Python

Python数据分析模块pandas用法详解

2019/09/04 Python

Python线程障碍对象Barrier原理详解

2019/12/02 Python

Python实现验证码识别

2020/06/15 Python

Python eval函数介绍及用法

2020/11/09 Python

size?瑞典：英国伦敦的球鞋精品店

2018/03/01 全球购物

Harrods美国：英国最大的百货公司

2018/11/04 全球购物

YBF Beauty官网：美丽挚友，美国知名彩妆品牌

2020/11/22 全球购物

优秀高中生事迹材料

2014/02/11 职场文书

小学班主任寄语大全

2014/04/04 职场文书

2014年餐厅服务员工作总结

2014/11/18 职场文书

2014年档案室工作总结

2014/12/01 职场文书

python 安全地删除列表元素的方法

2022/03/16 Python

进行数据处理的6个 Python 代码块分享

2022/04/06 Python

阿里云日志过滤器配置日志服务

2022/04/09 Servers