https://blog.csdn.net/m0_51030297/article/details/135141358
import torch
print(torch.cuda.is_available())
torch.cuda.current_device()
torch.cuda._initialized = True
将cuda._intialized设置为真
https://blog.csdn.net/m0_51030297/article/details/135141358
import torch
print(torch.cuda.is_available())
torch.cuda.current_device()
torch.cuda._initialized = True
将cuda._intialized设置为真