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

torch.cuda.device_count()
返回gpu数量；

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

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

Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

【2025最新】基于SpringBoot+微信小程序的微信云校园设计与实现mp管理系统源码+MyBatis+MySQL

avatar 2048 AI社区
cover

前后端分离物流管理系统系统|SpringBoot+微信小程序+MyBatis+MySQL完整源码+部署教程

avatar 2048 AI社区
cover

SpringBoot+微信小程序 小演员招募设计与开发_dabo平台完整项目源码+SQL脚本+接口文档【Java Web毕设】

avatar 2048 AI社区