当前位置：首页 > news >正文

PyTorch 2.0 以下版本中设置默认使用 GPU 的方法

news 来源：原创 2025/6/8 22:02:09

PyTorch 2.0 以下版本中设置默认使用 GPU 的方法

在 PyTorch 2.0以下版本中，默认情况下仍然是使用 CPU 进行计算，除非明确指定使用 GPU。在 PyTorch 2.0 以下版本中，虽然没有 torch.set_default_device 的便捷方法，但可以通过显式地将张量、模型和操作分配到 GPU 来使用 GPU。

1. 检查 GPU 可用性

在使用 GPU 之前，首先检查系统中是否有可用的 GPU。

import torch# 检查是否有可用的 GPU
print(torch.cuda.is_available())  # 返回 True 或 False# 检查可用 GPU 的数量
print(torch.cuda.device_count())# 当前 GPU 名称
if torch.cuda.is_available():print(torch.cuda.get_device_name(0))

2. 将张量移动到 GPU

张量可以通过 .to('cuda') 或 .cuda() 方法显式地移动到 GPU。

# 创建一个张量并将其移动到 GPU
x = torch.tensor([1.0, 2.0, 3.0])
x_gpu = x.to('cuda')  # 或 x.cuda()
print(x_gpu.device)  # 输出：cuda:0# 在 GPU 上进行计算
y = x_gpu * 2
print(y)  # 输出在 GPU 上的结果

3. 将模型移动到 GPU

PyTorch 模型及其参数需要显式地移动到 GPU。

# 定义一个简单的模型
model = torch.nn.Linear(10, 1)# 将模型移动到 GPU
model = model.to('cuda')  # 或 model.cuda()# 检查模型参数所在的设备
print(next(model.parameters()).device)  # 输出：cuda:0

4. 确保输入数据和模型在同一设备上

模型和输入数据需要在同一个设备上，否则会报错。

# 创建一个张量并移动到 GPU
input_data = torch.randn(5, 10).to('cuda')# 定义并移动模型到 GPU
model = torch.nn.Linear(10, 1).to('cuda')# 前向传播
output = model(input_data)
print(output)

5. 使用 `torch.device` 动态管理设备

可以使用 torch.device 动态管理设备。

# 定义设备
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')# 将张量移动到设备
x = torch.tensor([1.0, 2.0, 3.0]).to(device)# 将模型移动到设备
model = torch.nn.Linear(10, 1).to(device)

6. 优化器和损失函数的设备兼容性

当使用 GPU 时，模型的输出和目标（target）都需要在同一设备上。

# 创建数据和目标，并移动到 GPU
data = torch.randn(5, 10).to('cuda')
target = torch.randn(5, 1).to('cuda')# 定义模型并移动到 GPU
model = torch.nn.Linear(10, 1).to('cuda')# 定义损失函数
criterion = torch.nn.MSELoss()# 定义优化器
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)# 前向传播
output = model(data)
loss = criterion(output, target)# 反向传播
loss.backward()
optimizer.step()

7. 混合设备计算（可选）

在多 GPU 或混合 CPU/GPU 环境中，可以手动管理每个张量或模型的设备。

# 在 CPU 上创建张量
x_cpu = torch.tensor([1.0, 2.0, 3.0])# 在 GPU 上创建张量
x_gpu = x_cpu.to('cuda')# 将结果移动回 CPU
result = x_gpu * 2
result_cpu = result.to('cpu')
print(result_cpu)

总结

在 PyTorch 2.0 以下版本中，使用 GPU 的核心是 显式地将张量和模型移动到 GPU，并确保所有相关操作在同一设备上完成。以下是核心方法的汇总：

检查 GPU 可用性： torch.cuda.is_available()
移动张量到 GPU： .to('cuda') 或 .cuda()
移动模型到 GPU： .to('cuda') 或 .cuda()
动态设备管理： torch.device

PyTorch 2.0 以下版本中设置默认使用 GPU 的方法

PyTorch 2.0 以下版本中设置默认使用 GPU 的方法在 PyTorch 2.0以下版本中，默认情况下仍然是使用 CPU 进行计算，除非明确指定使用 GPU。在 PyTorch 2.0 以下版本中，虽然没有 torch.set_default_device 的便捷方法，但可以通过显式…...

编程日记 2025/6/4 17:20:53

Redis篇-19--运维篇1-主从复制（主从复制，读写分离，配置实现，实战案例）

1、概述 Redis的主从复制（Master-Slave Replication）是一种数据冗余机制，它允许将一台Redis服务器的数据复制到其他Redis服务器。在主从复制中，有一台主服务器（Master）和一个或多个从服务器（Sl…...

编程日记 2025/6/1 21:43:24

springboot449教学资源共享平台(论文+源码)_kaic

摘要如今社会上各行各业，都喜欢用自己行业的专属软件工作，互联网发展到这个时候，人们已经发现离不开了互联网。新技术的产生，往往能解决一些老技术的弊端问题。因为传统教学资源共享平台信息管理难度大，容错率低&am…...

编程日记 2025/5/30 21:14:18

Unbuntu下怎么生成SSL自签证书？

环境： WSL2 Unbuntu 22.04 问题描述： Unbuntu下怎么生成SSL自签证书？ 解决方案： 生成自签名SSL证书可以使用OpenSSL工具，这是一个广泛使用的命令行工具，用于创建和管理SSL/TLS证书。以下是生成自签名…...

编程日记 2025/6/2 7:32:18

Ubuntu18.04——换源

一、前提说明系统自带的源往往下载很慢，通过换源操作后，往往下载/更新速度大幅提升每种版本对应的不一样，例如Ubuntu18.04和Ubuntu20.04的有差异，所以换源需要根据不同版本对应的命令二、操作步骤 0.备份原先的 /etc/apt/sou…...

编程日记 2025/6/2 18:59:53

crictl和ctr与docker的命令的对比

crictl是遵循CRI接口规范的一个命令行工具，通常用它来检查和管理kubelet节点上的容器运行时和镜像 ctr是containerd的一个客户端工具， 接下来就是crictl的的常见命令，其中能完全替代docker命令的参照下列表格操作crictldocker查看运行容器…...

编程日记 2025/6/3 8:19:50

Java 技术面试常见问题解析

1.说说Mybatis的缓存机制: MyBatis 是一个优秀的持久层框架，它简化了企业应用开发中数据库操作的代码。MyBatis 提供了一级缓存和二级缓存机制来优化对数据库的访问。一级缓存 (SqlSession级别的缓存) 一级缓存是 MyBatis 中默认开启且无法关闭的缓存机制。它存…...

编程日记 2025/6/3 16:04:07

任意位置删除： void Any_Del(LinkListPtr h,int a)//任意删 {if(NULLh||a>h->len){printf("删除失败");}LinkListPtr ph;for(int i0;i<a-1;i){pp->next;}LinkListPtr p2p;p2p2->next;p->nextp->next->next;free(p2);p2NULL;h-&g…...

编程日记 2025/5/30 10:17:03

go引用包生成不了vendor的问题

比如我要引入github.com/jinzhu/gorm这个包. 1. 首先获取包 go get github.com/jinzhu/gorm 这时go.mod文件中也有这个包依赖信息了. 2. 然后构建vendor go mod vendor 结果发现vendor目录下没有生成对应的包, 而且modules.txt也注释掉这个包了. 原因是没有其进行引用, go…...

编程日记 2025/5/31 21:02:07

C语言——实现求出最大值

问题描述：利用C语言自定义函数求出一维数组里边最大的数字 //利用函数找最大数#include<stdio.h>int search(int s[9]) //查找函数 {int i , max s[0] , max_xia 0;for(i0;i<9;i){if(s[i] > max){max_xia i;max s[max_xia];}}return max; } in…...

编程日记 2025/6/2 6:26:16

【CSS in Depth 2 精译_081】 13.1：CSS 渐变效果（下）——CSS 径向渐变（13.1.3）+ CSS 锥形渐变（13.1.4）

当前内容所在位置（可进入专栏查看其他译好的章节内容） 第四部分视觉增强技术 ✔️【第 13 章渐变、阴影与混合模式】 ✔️ 13.1 渐变 ✔️ 13.1.1 使用多个颜色节点（上）13.1.2 颜色插值方法（中）13.1.3 径…...

编程日记 2025/6/2 22:05:22

【SH】Ubuntu Server 24搭建Web服务器访问Python程序研发笔记

文章目录说个问题写个方案一、安装Ubuntu Server二、安装Web服务器采用Nginx服务器三、安装Python及依赖创建项目虚拟环境四、安装Python Web框架采用Flask框架创建和运行Flask应用（以后的重点） 五、安装WSGI服务器采用Gunicorn 六、配置Nginx七、验证…...

编程日记 2025/6/1 6:28:16

创建项目以及本地仓库和远程仓库并上传项目

创建项目以及本地仓库和远程仓库并上传项目其详细流程如下： 1、本地创建项目 2、创建本地仓库（若使用idea在创建项目时选择了创建.git本地仓库，则此步骤省略） 进入到你需要上传的项目的目录下，右键找到Git Bah He…...

编程日记 2025/6/4 4:57:56

代码开发相关操作

使用Vue项目管理器创建项目：（vue脚手架安装一次就可以全局使用） windowR打开命令窗口，输入vue ui，进入GUI页面，点击创建-> 设置项目名称，在初始化git下面输入：init project&…...

编程日记 2025/6/3 6:31:15

ElasticSearch系列：利用runtime field实现日期字符串实现日期范围查询

在Elasticsearch中，如果你有一个时间字符串字段，并且你希望在查询时将其转换为date类型以便进行日期范围查询或其他日期相关的操作，你可以使用runtime_fields来实现这一转换。不过，与转换为UNIX时间戳不同，Elasticsear…...

编程日记 2025/5/30 19:15:12

前端：如何在静态目录下显示一张图片

假设已经配置（或默认配置好）public文件夹是静态资源文件夹，public文件夹中的资源会直接映射到根URL。 1. 我的前端图片保存路径是： F:\front\public\icon-favo.png 前端地址是：http://localhost:20002 我想要访问…...

编程日记 2025/6/3 0:44:35

Java设计模式 —— 【结构型模式】桥接模式详解

前言现在有一个需求，需要创建不同的图形，并且每个图形都有可能会有不同的颜色。首先我们看看用继承来实现： 我们可以发现有很多的类，假如我们再增加一个形状或再增加一种颜色，就需要创建更多的类。试想&#xf…...

编程日记 2025/5/30 20:03:29

Qt同步读取串口

头文件 #include "InsScpi.h" #include <QObject> #include <QSerialPort>class TestSerial : public QObject {Q_OBJECT public:explicit TestSerial(QObject *parent nullptr);//打开设备bool openDevice(const QString &portName);//关闭设备…...

编程日记 2025/5/31 16:22:48

MySQL高可用

MySQL主从复制的过程是怎么样的分为3个阶段： 写入binlog：主库修改数据后，会写入binlog日志，从库连接到主库后，主库会创建一个log dump线程，用于发送bin log的内容同步binlog：从库会专门创建一…...

编程日记 2025/6/4 1:52:23

OpenHarmony-3.HDF Display子系统（6）

Display 子系统 1.Display驱动模型介绍当前操作系统和 SOC 种类繁多，各厂商的显示屏器件也各有不同，随之针对器件的驱动代码也不尽相同，往往是某一款器件驱动，只适用于某单一内核系统或 SOC，如果要迁移到其他内核或者…...

编程日记 2025/6/2 20:33:29

第10章：CSS最佳实践 --[CSS零基础入门]

代码组织在CSS开发中，良好的代码组织和最佳实践对于项目的可维护性和扩展性至关重要。以下是两个示例，展示了如何遵循CSS最佳实践来组织代码。示例 1: 使用 BEM（Block Element Modifier）命名法 BEM 是一种用于提高 CSS 可读性…...

编程日记 2025/5/31 5:15:30

备战美赛！2025美赛数学建模C题模拟预测！用于大家练手模拟！

完整的思路代码模型见文末 2025 美赛数学建模 C 题模拟题：城市交通拥堵指数的预测与管理策略背景随着全球城市化进程的加快，交通拥堵问题成为城市发展的重要挑战之一。交通拥堵不仅影响居民出行效率，还增加了能源消耗和碳排放。近年来&…...

编程日记 2025/6/1 19:26:13

ESP8266 Ubuntu 安装

文章参考：https://blog.csdn.net/AUST_129/article/details/119406722文章浏览阅读1.8k次，点赞4次，收藏19次。参考：https://docs.espressif.com/projects/esp8266-rtos-sdk/en/latest/get-started/linux-setup.htmlhttp://aicloud…...

编程日记 2025/5/30 0:54:55

tryhackme-Pre Security-Defensive Security Intro（防御安全简介）

任务一：Introduction to Defensive Security防御安全简介此room的两个要点： Preventing intrusions from occurring 防止入侵发生Detecting intrusions when they occur and responding properly 检测发生的入侵并正确响应防御安全还有更多内容。除上…...

编程日记 2025/6/3 6:42:59

单片机：实现倒计时（附带源码）

使用单片机实现倒计时功能是一个常见的嵌入式应用，它能帮助你更好地理解如何进行时间控制和如何通过定时器实现精确的倒计时。通过该项目，你将学习如何使用单片机的定时器来进行时间计算，并通过LED或LCD显示倒计时的结果。 1. 项目概述倒计…...

编程日记 2025/6/3 1:32:03

安全防御之可信计算技术

可信计算技术是一种计算机安全体系结构，旨在提高计算机系统在面临各种攻击和威胁时的安全性和保密性。它通过包括硬件加密、受限访问以及计算机系统本身的完整性验证等技术手段，确保计算机系统在各种攻击和威胁下保持高度安全和保密性。一、可信计算基…...

编程日记 2025/5/29 13:23:46

视频生成缩略图

文章目录视频生成缩略图使用ffmpeg 视频生成缩略图最近有个需求，视频上传之后在列表和详情页需要展示缩略图使用ffmpeg 首先引入jar包 <dependency><groupId>org.bytedeco</groupId><artifactId>javacpp</artifactId><vers…...

编程日记 2025/6/3 20:41:36

PySide6程序框架设计

pyside6有一个优点自动适配高分辨ui pyqt5需要自己写这部分逻辑 1、主程序代码 DINGSHI01Main.py # -*- coding: utf-8 -*- import sys,time,copy from PySide6.QtWidgets import QWidget,QApplication from PySide6.QtCore import Qt from PySide6 import QtCore, QtGui, Q…...

编程日记 2025/6/4 1:52:19

WebSocket入门与结合redis

WebSocket是什么 WebSocket 是一种用于在客户端和服务器之间建立双向通信的协议，它能实现实时、持久的连接。与传统的 HTTP 请求响应模式不同，WebSocket 在建立连接后允许客户端和服务器之间相互发送消息，直到连接关闭。由于 WebSocket 具有…...

编程日记 2025/5/29 14:24:40

锂电池SOH预测 | 基于BiGRU双向门控循环单元的锂电池SOH预测，附锂电池最新文章汇集

锂电池SOH预测 | 基于BiGRU双向门控循环单元的锂电池SOH预测，附锂电池最新文章汇集目录锂电池SOH预测 | 基于BiGRU双向门控循环单元的锂电池SOH预测，附锂电池最新文章汇集预测效果基本描述程序设计参考资料预测效果基本描述锂电池SOH预测 | 基于Bi…...

编程日记 2025/6/3 21:26:49

C# 结构体和类

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档文章目录前言一、类（Class）二、结构体（Struct）示例代码（定义类和结构体）类的继承代码示例&#xff08…...

编程日记 2025/6/2 10:13:35

C语言中的内存管理：理解指针、动态内存分配与内存泄漏

在C语言中，内存管理是一个至关重要的主题。与许多高级语言不同，C语言要求程序员显式地管理内存的分配与释放。虽然这种做法提供了更高的灵活性和控制权，但也容易导致内存泄漏、越界访问等问题。正确地管理内存对于编写高效、稳定的C程序至关重…...

编程日记 2025/6/2 2:57:56

web：pc端企业微信登录-vue版

官方文档：developer.work.weixin.qq.com/document/pa… 不需要调用ww.register，直接调用ww.createWWLoginPanel即可创建企业微信登录面板 - 文档 - 企业微信开发者中心 (qq.com) 引入 //通过 npm 引入 npm install wecom/jssdk import * as ww from we…...

编程日记 2025/6/3 13:26:11

GC.2015.四年级

GC.2015.四年级.01.奖励题目描述晨晨班主任想奖励班里面的每个学生一只圆珠笔和铅笔，已知每只圆珠笔和铅笔的价格，以及班里面的学生人数n，你能帮助老师算出总价吗？ 输入格式第一行：一个整数n，代表班里…...

编程日记 2025/6/4 7:22:40

一篇文章掌握WebService服务、工作原理、核心组件、主流框架

目录 1、WebService定义解决问题： 2、WebService的工作原理 2.1 实现一个完整的Web服务包括以下步骤 2.2 调用方式 3、Web Service的核心组件 3.1 XML 3.2 SOAP 3.3 WSDL 3.4 UDDI 4、主流框架 4.1 AXIS(已淘汰) 4.2 XFire 4.3 CXF 5、Soap协议详解…...

编程日记 2025/6/3 6:37:25

中软高科身份证云解码金融（银行）解决方案介绍

多年来，中软高科一直深耕身份证云解码领域，对身份证云解码应用于金融（银行），进行了大量且深入的研究。从长期调研来看，金融（银行）的痛点需求主要有： 传统身份证解码设备…...

编程日记 2025/6/3 12:55:29

Linux NVIDIA GPU linpack 测试

前言多节点多GPU测试有点坑，这篇文章有解决方法。环境操作系统信息 lsb_release -aNo LSB modules are available. Distributor ID: Ubuntu Description: Ubuntu 22.04.3 LTS Release: 22.04 Codename: jammycpu 信息 lscpuArchitecture: x86_64CPU op-mod…...

编程日记 2025/6/1 20:57:03

LiteFlow决策系统的策略模式，顺序、最坏、投票、权重

个人博客：无奈何杨（wnhyang） 个人语雀：wnhyang 共享语雀：在线知识共享 Github：wnhyang - Overview 想必大家都有听过或做过职业和性格测试吧，尤其是现在的毕业生，在投了简历之后经…...

编程日记 2025/6/3 19:31:24

“AI换脸”骗过人脸识别？黑产攻击新手段应如何防御？

在著名美剧《权力的游戏》中，有一个神秘的刺客组织叫“无面者”，这个组织中的人可以通过某种神秘手段切换无数种不同的面孔，实现“一人千面”。电视剧毕竟魔幻。但如今，基于人工智能的深度合成伪造技术正在让“一人千面”成为现…...

编程日记 2025/6/3 8:46:56

面试题整理6----什么是进程最大数、最大线程数、进程打开的文件数，怎么调整

什么是进程最大数、最大线程数、进程打开的文件数，怎么调整 1. 进程最大数1.1 调整方法： 2. 最大线程数2.1 调整方法： 3. 注意事项 #linux 1. 进程最大数进程最大数是指操作系统允许同时运行的进程数量上限。这个限制通常由内核参数 ulimi…...

编程日记 2025/5/29 8:40:46

android RadioButton + ViewPager+fragment

RadioGroup viewpage fragment 组合显示导航栏 1、首先主界面的布局控件就是RadioGroup viewpage <?xml version"1.0" encoding"utf-8"?> <LinearLayout xmlns:android"http://schemas.android.com/apk/res/android"xmlns:tools…...

编程日记 2025/6/1 10:38:39

Zabbix6.0升级为6.4

为了体验一些新的功能，比如 Webhook 和问题抑制等，升级个小版本。一、环境信息 1. 版本要求一定要事先查看官方文档，确认组件要求的版本，否则版本过高或者过低都会出现问题。 2. 升级前后信息环境升级前升级后操作系统CentOS…...

编程日记 2025/5/28 20:14:00

Unity 根据文本宽度自动移动图像位置

游戏中有时候需要变动的显示一个物品的数量，变化的文本宽度不停的变化，这时候需要将物品的icon随着文本的长度而改变位置。实现思路：使用Content Size Fitter来动态改变内容的大小。首先建立一个文本组件，添加Content Size Fi…...

编程日记 2025/6/4 4:42:08

spring @Mapper Converter转换泛型异常

spring Mapper Converter转换泛型异常需要在每个list类型转换上面加Named 注解，否则会影响page生成的类型转换比如： import org.mapstruct.Mapper; import org.mapstruct.Named;import com.baomidou.mybatisplus.core.metadata.IPage; import com.b…...

编程日记 2025/5/30 23:22:35

如何设计一个秒杀系统

开局一张图结局要说清对于设计一个秒杀系统，结合图片分层结构，根据每一层从访问层，负载层，服务层，业务层，支撑层，数据层，详细说明每一层应该怎么设计。应该注意那些事项。比如访…...

编程日记 2025/6/4 16:31:05

SPL06 基于stm32F103 HAL库驱动（软件模拟IIC）

talk is cheap, show you my code SPL06.c #include "SPL06.h"//*************全局变量*************// Factor_List* b_list; //存储过采样率对应的系数KP，KT COEF_ValueStruct Coefficient { 0 }; //存储校准系数…...

编程日记 2025/5/31 15:37:57

arcgisPro将面要素转成CAD多段线

1、说明：正常使用【导出为CAD】工具，则导出的是CAD三维多线段，无法进行编辑操作、读取面积等。这是因为要素面中包含Z值，导出则为三维多线段数据。需要利用【复制要素】工具禁用M值和Z值，再导出为CAD，则得到…...

编程日记 2025/5/31 8:23:17

Cocos Creator 试玩广告开发

之前主要是使用Unity,这次刚好项目是试玩游戏的开发，所以临时学了Cocos来开发。所以这篇文章，更加关注从Unity转到Cocos开发的经历以及试玩的基本开发。首先，我是没有使用过Cocos的，也没有接触过Ts语言，对于Ts的开发开…...

编程日记 2025/5/31 10:30:02

【Linux】解锁文件描述符奥秘，高效缓存区的实战技巧

fd和缓冲区 1. 文件描述符fd1.1. 概念与本质1.2. 打开文件的管理1.3. 一切皆文件的理解1.4. 分配规则1.5. 重定向的本质1.5.1. dup2 2. FILE中的缓冲区2.1. 概念2.2. 存在的原因2.3. 类型(刷新方案)2.4. 存放的位置2.4.1. 代码证明、现象解释 2.5. 模拟C标准库中的方法 1. 文件…...

编程日记 2025/6/2 4:31:17

MySQL基础笔记（五）

在此特别感谢尚硅谷-康师傅的MySQL精品教程获取更好的阅读体验请前往我的博客主站! 如果本文对你的学习有帮助，请多多点赞、评论、收藏，你们的反馈是我更新最大的动力！ 约束 1. 约束(constraint)概述 1.1 为什么需要约束数据完整性&…...

编程日记 2025/6/4 18:03:12

PyTorch 2.0 以下版本中设置默认使用 GPU 的方法

1. 检查 GPU 可用性

2. 将张量移动到 GPU

3. 将模型移动到 GPU

4. 确保输入数据和模型在同一设备上

5. 使用 torch.device 动态管理设备

6. 优化器和损失函数的设备兼容性

7. 混合设备计算（可选）

总结

相关文章：

5. 使用 `torch.device` 动态管理设备