查看Pytorch是否使用GPU

下面的命令可以帮助我们查看Pytorch是否使用GPU：# 返回当前设备索引torch.cuda.current_device()# 返回GPU的数量torch.cuda.device_count()# 返回gpu名字，设备索引默认从0开始torch.cuda.get_device_name(0)# cud...

花开山岗红艳艳

3746人浏览 · 2021-04-18 11:37:21

花开山岗红艳艳 · 2021-04-18 11:37:21 发布

下面的命令可以帮助我们查看Pytorch是否使用GPU：

# 返回当前设备索引
torch.cuda.current_device()

# 返回GPU的数量
torch.cuda.device_count()

# 返回gpu名字，设备索引默认从0开始
torch.cuda.get_device_name(0)

# cuda是否可用
torch.cuda.is_available()

我的结果是：

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Bj0dLYom-1609642835880)(C:\Users\Administrator\Desktop\查看pytorch是否使用GPU.png)]$

转载自：https://blog.csdn.net/weixin_43593330/article/details/112132428

昇腾开源生态专区

昇腾计算产业是基于昇腾系列（HUAWEI Ascend）处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务，https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐

Flutter 框架跨平台鸿蒙开发 - 开发二维码生成器与扫描器

size: 200,// 定位点样式（三个角的方块）eyeShape: QrEyeShape.square, // square 或 circle),// 数据点样式dataModuleShape: QrDataModuleShape.square, // square 或 circle),二维码生成：qr_flutter库的使用和样式自定义二维码扫描：mobile_scanner相机调用和结果处

昇腾开源生态专区

必学收藏！vLLM-Omni全模态模型高效推理引擎：从入门到实践

昇腾开源生态专区

PD分离部署：基于Atlas 800T A3服务器+SGLang 1P2D部署Qwen3-235B指南

Qwen3-235B模型用Atlas 800I A3或Atlas 800T A3均可部署，本文档以Atlas 800T A3为例，记录了使用SGLang框架部署Qwen3-235B模型的1P2D（一个Prefill节点和两个Decode节点）配置方案。该配置针对大规模语言模型推理场景进行了优化，特别适用于需要高吞吐量的生产环境。