[pytorch]查看GPU信息

20181128 qzd

https://pytorch.org/docs/stable/cuda.html

  1. 查看GPU是否可用
torch.cuda.is_available()
  1. 返回GPU数量
torch.cuda.device_count()
  1. 返回GPU名字，设备索引默认从0开始
torch.cuda.get_device_name(0)
  1. 返回当前设备索引
torch.cuda.current_device()
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容

  • 与 TensorFlow 的初次相遇
    与 TensorFlow 的初次相遇 https://jorditorres.org/wp-content/upl...
    布客飞龙阅读 3,970评论 2赞 89
  • PyTorch 深度学习: 60分钟快速入门
    作者:Soumith Chintala 官方60分钟快速入门翻译 Github 地址简书地址CSDN地址 本教程的...
    MaosongRan阅读 25,992评论 0赞 35
  • CUDA——"从入门到放弃"
    开篇一张图，后面听我编 1. 知识准备 1.1 中央处理器（CPU） 中央处理器（CPU，Central Proc...
    He_Yu阅读 47,425评论 7赞 115
  • 絮絮叨叨第一篇
    最近睡的都特别好，可能是因为开了空调的缘故。南京的温度居高不下，每天顶着三十度的高温奔走在快融化的马路上，热气从地...
    憧彼淮夷阅读 201评论 0赞 0
  • 为你我冷漠
    词/花海洋 轻轻的你说有个午后我说了谎话 因此你走远远离我 默默的我追寻二十年前那个午后 当时电话厅里接到一个电话...
    花海洋阅读 411评论 2赞 8