检查cuda和显卡的可用性
检查cuda和显卡的可用性
import torch
device_gpu = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(device_gpu)
print(torch.cuda.is_available())
原文地址：https://blog.csdn.net/qq_42830971/article/details/142693431
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！