PyTorch默认用CPU,改默认用GPU,import torch
后,用torch.set_default_device('cuda')
改为默认使用GPU
PyTorch改为默认使用GPU
阅读 12
2024-09-07
PyTorch默认用CPU,改默认用GPU,import torch
后,用torch.set_default_device('cuda')
改为默认使用GPU
相关推荐
精彩评论(0)