1 import torch 2 import torchvision 3 print(torch.cuda.is_available())
pytorch查看CUDA支持情况,只需要三行代码
猜你喜欢
转载自www.cnblogs.com/liuke-note/p/10149530.html
今日推荐
周排行