深度学习:测试是否含有gpu的环境
摘要: 在我们跑深度学习的时候,绕不开的一定是gpu显卡,但是如何检测是否使用gpu那?下面我讲提供一段代码,可以检测自己是否开启了gpu或含有gpu。 import torch # 检查是否支持 CUDA(即是否有可用的 NVIDIA GPU) flag = torch.cuda.is_available
9
0
0
点击右上角即可分享
微信分享提示
深色
回顶
收起
  1. 1 倔强 五月天
倔强 - 五月天
00:00 / 00:00
An audio error has occurred.