当前位置: 首页 > 知识库问答 >
问题:

如何使用pytorch列出所有当前可用的GPU?

汪天宇
2023-03-14

我知道我可以使用torch访问当前的GPU。库达。当前的_device(),但如何获取所有当前可用GPU的列表?

共有1个答案

锺离自明
2023-03-14

您可以通过执行以下操作列出所有可用的GPU:

>>> import torch
>>> available_gpus = [torch.cuda.device(i) for i in range(torch.cuda.device_count())]
>>> available_gpus
[<torch.cuda.device object at 0x7f2585882b50>]
 类似资料:
  • 问题内容: 我想知道是否可以使用JNI api列出当前JVM中的所有当前可用实例(作为jobject)。 我的意思示例: 我的任务是在它们中搜索实现特定interface()的对象,我必须动态且全局地执行此操作,而无需类名 问题答案: JVMTI将提供帮助。 调用以标记所有必需的对象; 调用将所有标记的对象复制到数组。 这是一个例子。注意,它也可以是接口。

  • 问题内容: 例如,当我想设置在 谢谢。 问题答案: 检查这其他的选择。

  • 问题内容: 我想建立一个基于php的站点,该站点在我的Ubuntu服务器上(自动)执行一些命令 我要做的第一件事是转到文件(sudoers)并添加用户www-data,以便我可以以root权限执行php命令! 然后我的PHP代码是 它仅返回一个用户(最后一个用户)!如何使其返回所有用户? 谢谢 问题答案: 从exec的PHP手册中: 返回值 命令结果的最后一行。如果您需要执行命令并直接将命令中的所

  • 问题内容: 我正在使用ResourceBundle,我想给用户一个选择GUI语言的选项。 我想获取特定程序包下所有资源文件的列表。 我不知道我将拥有什么资源,因为此应用程序基于插件。 是否有一个选项可以要求Java搜索程序包下的所有可用资源? (如果没有,我猜该插件应该为它提供所有可用的本地文件) 谢谢你们 问题答案: 这些文件可以驻留在Web服务器上。没有标准的方式列出Web服务器上的文件(页面

  • 问题内容: 如何列出Linux中某个组的所有成员(可能还包括其他unice)? 问题答案: 不幸的是,据我所知,没有很好的便携式方法可以做到这一点。如果您尝试解析/ etc / group(如其他人所建议的那样),则会错过以该组为主要组的用户以及通过UNIX平面文件(例如LDAP,NIS, pam-pgsql等)。 如果我绝对必须自己做,则可能相反:使用来获取系统上每个用户的组(这将使NSS看到所

  • 我正在写一个shell脚本来监视Kafka经纪人。 我浏览了一些链接,发现如果ZooKeeper包含一个代理列表,如果这个列表中有IP地址,那么一个kafka代理正在运行。 我想要一个命令,我可以在我的shell脚本中使用,以获得代理列表,并检查kafka是否正在运行。 是否有命令可以像ElasticSearch一样获取kafka集群状态?