pytorch 查看cuda 版本方式
作者：峰峰晶晶 发布时间：2022-07-16 23:43:05
标签：pytorch,cuda
由于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)
来源：https://blog.csdn.net/ewqapple/article/details/95067814
0
投稿猜你喜欢
前言最近因为工作需要 用selenium做了一个QQ邮箱的爬虫(登录时部分帐号要滑动解锁),先简单记录一下。这个问题先可以分为两个部分:1.- 本文实例讲述了Python使用ConfigParser模块操作配置文件的方法。分享给大家供大家参考，具体如下：一、简介用于生成和修改常见配置
- 首先，与其他语言不同，JS的效率很大程度是取决于JS engine的效率。除了引擎实现的优劣外，引擎自己也会为一些特殊的代码模式采取一些优化
- 经常看见MOP上有人贴那种动态的图片，就是把一个字符串作为参数传给一个 * 页，就会生成一个带有这个字符串的图片，这个叫做文字水印。像什么原
- 目录1、连接MongoDB2、指定（切换）数据库3、指定（切换）集合4、插入数据5、查询数据6、统计查询7、结果排序8、偏移9、更新数据10
- 前段时间自学了python，作为新手就想着自己写个东西能练习一下，了解到python编写爬虫脚本非常方便，且最近又学习了MongoDB相关的
- 平面设计 常用尺寸 三折页广告 标准尺寸: (A4)210mm x 285mm普通宣传册 标准尺寸: (A4)210mm x 285mm文件
- 本文主要讲述的是Python中的模块的概念，具体如下。模块是python组织代码的基本方式：python的脚本都是用扩展名为py的文本文件保
- 前一篇博客说了怎样通过命名管道实现进程间通信，但是要在windows是使用命名管道，需要使用python调研windows api，太麻烦，
- 参考网上的例子，实现了简单的matplotlib pyqt5绘图 相关知识点： （1）pyqt5中添加控件要在布局中添
- 在过去的两年里，你能想象的一些很烂的应用设计也入驻了优雅的iPhone。通过它周边的宣传，使得世界各地的设计者们都利用这个新的移动工具来一展
- 在开发的过程中，我们不可避免的会遇到各种各样的编码，解码，或者乱码问题，很多时候，我们可以正常的解决问题，但是说实在的，我们有可能并不清楚问
- sympy版本:1.2假设求解矩阵方程AX=A+2X其中求解之前对矩阵方程化简为(A−2E)X=A令B=(A−2E)使用qtconsole输
- 1. 什么是XSLT 大家可能听说过XSL(eXtensible Stylesheet Language)，XSL和我们这里说的XSLT从狭
- 介绍还记得你在小学时学习如何加减数字吗？现在，你也可以对图像做同样的事情!输入图像可以进行算术运算，例如加法、减法和按位运算（AND、OR、
- 场景可能是你用不到，但是我遇到了这样一个问题，就是我想详细了解我的竞争对手的网站（电商类）销售情况和新品上架情况，但是我总不至于像盯盘一样，
- 前言利用Python实现获取动态图表，废话不多说~让我们愉快地开始吧~开发工具Python版本： 3.6.4相关模块：re模块；reques
- 查询一天：select * from table where to_days(column_time) = to_days(now());s
- 对于非连续数据集，数据可视化时候需要每七天一个采样点。要求是选择此前最新的数据作为当日的数据展示，譬如今天是2019-06-18，而数据集里
- 一、了解subprocesssubeprocess模块是python自带的模块，无需安装，主要用来取代一些就的模块或方法，如os.syste
