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
投稿猜你喜欢
本文实例介绍了基于python的Tkinter实现简易计算器的详细代码，分享给大家供大家参考，具体内容如下第一种：使用python 的 Tk- 与上篇实践教程一样，在这篇文章中，我将继续从一种常见的功能——表格入手，展示Vue.js中的一些优雅特性。同时也将对filter功能与com
- 本文实例讲述了Python实现删除文件中含指定内容的行。分享给大家供大家参考，具体如下：#!/bin/env pythonimport sh
- 本文实例为大家分享了python+opencv识别图片中足球的方法，供大家参考，具体内容如下先补充下霍夫圆变换的几个参数知识：dp，用来检测
- 可匹配单行，也支持换行匹配[\s\S]*?加上括号，效果更好([\s\S]*?)来源：https://blog.csdn.net/ASUKA
- 在本教程中使用MySQL最新的MySQL服务8.0.11的社区绿色版本进行安装，绿色版为zip格式的包，安装分为以下四步走：第一步：下载My
- 一、前言在写业务代码时候，有许多场景需要重试某块业务逻辑，例如网络请求、购物下单等，希望发生异常的时候多重试几次。本文分享如何利用Pytho
- 不知道怎么忽然想看这个,呵呵 小我的python的反shell的代码 #!/usr/bin/python # Python&nbs
- 我们在做网站的时候，网站后台系统一般都会用到web编辑器，今天笔者就给大家推荐一款百度UEditor编辑器。关于这款百度UEditor编辑器
- 【目标需求】查看某一个字符串是否为中文字符串【解决办法】def check_contain_chinese(check_str): for
- python中的Reportlab模块reportlab模块是用python语言生成pdf文件的模块安装：pip install repor
- 如何在线创建新表？下面我们以建立一个数码相机库用表为例，看看在ASP程序代码中使用 [CREATE TABLE 相机 (品牌 TEXT(10
- 利用zipfile模块和pandas获取数据,代码比较简单，做个记录吧：# -*- coding: utf-8 -*-""
- 0. 简介上一篇博客简单介绍了GMP模型，这一篇我们介绍一下Go调度器的初始化过程，也就是在main.main函数运行之前所做的事情。1.
- 这篇文章主要介绍了python处理RSTP视频流过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的
- 本文实例讲述了JS实现密码框根据焦点的获取与失去控制文字的消失与显示效果。分享给大家供大家参考，具体如下：思路：1、首先用把密码框用txt暂
- 在JAVA WEB应用中，如何获取servlet请求中的参数，并传递给跳转的JSP页面？例如访问http://localhost:8088/
- 前言django，web开发中，用django-debug-toolbar来调试请求的接口，无疑是完美至极。 可能本人，见识博浅，才说完美至
- 本文实例讲述了python简单获取本机计算机名和IP地址的方法。分享给大家供大家参考。具体实现方法如下：方法一：>>> i
- 简介在之前的编程中，我们的信息打印，数据的展示都是在控制台（命令行）直接输出的，信息都是一次性的没有办法复用和保存以便下次查看，今天我们将学
