问题描述
使用CUDA进行模型训练时出现错误
RuntimeError: No CUDA GPUs are available
问题解决
使用CUDA之前检测是否有GPU,没检测就默认没有
print(torch.cuda.is_available())
使用CUDA进行模型训练时出现错误
RuntimeError: No CUDA GPUs are available
使用CUDA之前检测是否有GPU,没检测就默认没有
print(torch.cuda.is_available())