pytorch查看CUDA支持情况,只需要三行代码

1  import torch
2  import torchvision
3  print(torch.cuda.is_available())

猜你喜欢

转载自www.cnblogs.com/liuke-note/p/10149530.html