网站首页
编程语言
数据库
系统相关
其他分享
编程问答
GPUPyTorch
2024-09-07
PyTorch改为默认使用GPU
PyTorch默认用CPU，改默认用GPU，importtorch后，用torch.set_default_device('cuda')改为默认使用GPU