PyTorch默认用CPU,改默认用GPU,import torch
后,用torch.set_default_device('cuda')
改为默认使用GPU
PyTorch默认用CPU,改默认用GPU,import torch
后,用torch.set_default_device('cuda')
改为默认使用GPU
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M