问题:torch.cuda.is_available()为False
查看当前电脑cuda是否可用,代码:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)
输出:
1.7.1+cpu
False
None
微信扫一扫
查看当前电脑cuda是否可用,代码:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)
输出:
1.7.1+cpu
False
None
相关推荐