Pytorch如何切换 cpu和gpu的使用详解
作者：遇到好事了 发布时间：2023-08-22 03:30:11
标签：Pytorch,切换,cpu,gpu
前言，在pytorch中，当服务器上的gpu被占用时，很多时候我们想先用cpu调试下代码，那么就需要进行gpu和cpu的切换。
方法1：x.to(device)
把 device 作为一个可变参数，推荐使用argparse进行加载：
使用gpu时：
device='cuda'
x.to(device) # x是一个tensor，传到cuda上去
使用cpu时：
device='cpu'
x.to(device)
方法2：使用x.cuda()+CUDA_VISIBLE_DEVICES
很多贴子中说，使用x.cuda() 和x.to('cuda') 虽然是等效的，但是x.cuda() 的缺点是无法动态切换cpu。然而，其实配合命令行参数CUDA_VISIBLE_DEVICES 是可以进行切换的。
在服务器上创建一个python脚本 t.py：
import torch
print(torch.cuda.device_count()) # 可用gpu数量
print(torch.cuda.is_available()) # 是否可用gpu
首先先看一下，正常运行的情况：
执行命令：python t.py
输出结果：因为服务器上有两个gpu，所以是我们想要的结果。
2
True
如果想要只使用某一块gpu，只需要在执行前加一个参数：
CUDA_VISIBLE_DEVICES=0 python t.py，例如，我们要使用gpu 0
接下来看看输出什么：是的！程序中确实只可见了一块gpu~
1
True
下面，如果我们想使用cpu呢？
CUDA_VISIBLE_DEVICES="" python t.py
输出结果：可以看到，虽然服务器上有2块cpu，通过我们设置执行参数，程序中也成功看不到了！
0
False
因此，回归正题，当我们使用x.cuda()进行分配gpu时，只需要使用torch.cuda.is_available()加一个判断即可，当想使用cpu的时候在执行程序的命令行参数进行控制：
if torch.cuda.is_available():
x= x.cuda()
来源：https://blog.csdn.net/qq_42549774/article/details/114192405
0
投稿猜你喜欢
- python中类的继承：子类继承父类，及子类拥有了父类的 属性 和 方法。python中类的初始化都是__init__()。所以父类和子类的
python注释方法方式1单行注释：shift + #（在代码的最前面输入，非选中代码进行注释）多行注释：同单行一样在每一行的前面输入shi- 很久没有更新blog了，这段时间实在是发生了很多的事，累身累心。但还是有很多想做的事，比如更新merceCSS、把一直以来所总结的有关模块化
- 这篇文章主要介绍了如何使用python实现模拟鼠标点击,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋
- 上下文管理器是一种 Python 构造，它提供了一个类似 try-finally 的环境，具有一致的接口和方便的语法，例如通过&ld
- 一般来说，我们会将自己写的Python模块与python自带的模块分开存放以达到便于维护的目的。那么如何在Python中添加自定义的模块呢？
- 1 为什么找不见外星人为什么我们见不到外星人？ 曾经在物理学上有一个著名人物叫费米，大家知道费米是在物理学上发现中子轰击的人，有一个著名的费
- 一、数字类型所谓的“数字类”，就是指 DECIMAL 和 NUMERIC，它们是同一种类型。它严格的
- 要想从命令行启动mysqld服务器，你应当启动控制台窗口(或“DOS window”)并输入命令：C
- 目录1. 什么是闭包2. 闭包的作用2.1) 记忆性2.2) 模拟私有变量3. 闭包的注意点总结1. 什么是闭包闭包：函数本身和该函数声明时
- mysql安装好经常发现无法正常启动碰到最多的是error 2003的错误，以下为解决方法： mysqld -nt -remove mysq
- 用下面代码可实现：<％Dim writeDim fileSysObj, tf, readrea
- 前阵子刚完成一个B/S架构的学校办公系统，体会就是表太多，文件太多，而每个文件中类似的操作（代码）也太多了，例如学生信息和教师信息操作，st
- 记录了mysql 8.0.12下载安装教程，分享给大家。下载下载地址如图下载以后将安装包解压到任意文件夹，我这里解压到E盘。安装1、解压以后
- 一、简介 XML（eXtensible Markup Languag
- 关于break/continue这两个关键字在平常的使用过程中一直比较迷糊。好不容易理解了吧，过段时间不使用好像忘记了什么。这个问题也是很多
- <?php /** 默认首页 **/ class DefaultController extends AppController {
- 本文实例为大家分享了Python将图片彩色转化为素描的具体代码，供大家参考，具体内容如下第一种：from PIL import Image,
- 编辑距离编辑距离（Edit Distance），又称Levenshtein距离，是指两个字串之间，由一个转成另一个所需的最少编辑操作次数。编
- 1.列表元素删操作的方法列表的删操作指的是在列表中删除已存在的元素，列表中的元素被删除后，后面所有的元素依次往前移动一位，挂在被删除元素的索
