Loading

2025年2月10日
深度学习：测试是否含有gpu的环境
摘要： 在我们跑深度学习的时候，绕不开的一定是gpu显卡，但是如何检测是否使用gpu那？下面我讲提供一段代码，可以检测自己是否开启了gpu或含有gpu。 import torch # 检查是否支持 CUDA（即是否有可用的 NVIDIA GPU） flag = torch.cuda.is_available 阅读全文
posted @ 2025-02-10 10:03 虾饺爱下棋 阅读(186) 评论(0) 推荐(0)