验证gpu版pytorch是否可用
生活随笔
收集整理的這篇文章主要介紹了
验证gpu版pytorch是否可用
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
import torchtorch.cuda.is_available()
Truetorch.cuda.current_device()
0torch.cuda.device(0)
<torch.cuda.device object at 0x0000023034332438>torch.cuda.device_count()
1torch.cuda.get_device_name(0)
'GeForce GTX 1660'
?
總結
以上是生活随笔為你收集整理的验证gpu版pytorch是否可用的全部內容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： C++查看各种数据类型所占字节和最大最小
- 下一篇： db2时间格式化为时间到时分秒_快捷指令