torch查看GPU信息常用代码

python 复制代码
torch.cuda.device_count() # 可以返回可用 GPU 的数量。

torch.cuda.get_device_name(device) # 可以获取特定 GPU 的名称。其中 device 是 GPU 的索引(从 0 开始)

# 查看所有gpu
for i in range(torch.cuda.device_count()):
    print(torch.cuda.get_device_name(i))
'''
Tesla T4
Tesla T4
'''
相关推荐
新智元5 分钟前
AI是「天才」还是「话术大师」?Anthropic颠覆性实验,终揭答案!
人工智能·openai
TG:@yunlaoda360 云老大7 分钟前
2025云栖大会举行:阿里云旗舰模型Qwen3-Max、下一代架构Qwen3-Next重磅发布,加速构建“云智一体”AI超级计算机
人工智能·阿里云·架构
华为云开发者联盟9 分钟前
【新特性】 版本速递 | 华为云Versatile智能体平台 新增特性介绍(2025年10月发布)
人工智能·华为云开发者联盟·ai agent·mcp·华为云versatile
Francek Chen26 分钟前
【自然语言处理】预训练02:近似训练
人工智能·pytorch·深度学习·自然语言处理
云和数据.ChenGuang29 分钟前
tensorflow的广播机制
人工智能·python·tensorflow
J_Xiong011736 分钟前
【VLNs篇】11:Dynam3D: 动态分层3D令牌赋能视觉语言导航中的VLM
人工智能·算法·3d
极客BIM工作室39 分钟前
AI 图像生成技术发展时间脉络:从 GAN 到多模态大模型的知名模型概略解析
人工智能·神经网络·生成对抗网络
学术小白人1 小时前
11月即将召开-IEEE-机械制造方向会议 |2025年智能制造、机器人与自动化国际学术会议 (IMRA 2025)
运维·人工智能·机器人·自动化·制造·rdlink研发家
碧海银沙音频科技研究院1 小时前
i2s封装成自己定义8路音频数据发送方法
arm开发·人工智能·深度学习·算法·音视频
做科研的周师兄1 小时前
【机器学习入门】9.2:感知机的工作原理 —— 从模型结构到实战分类
人工智能·算法·机器学习·分类·数据挖掘