torch.cuda.is_available()
cuda是否可用；
torch.cuda.device_count()
返回gpu數量；
torch.cuda.get_device_name(0)
返回gpu名字，設備索引默認從0開始；
torch.cuda.current_device()
返回當前設備索引；
pytorch中查看gpu信息
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。