print(torch.cuda.is_available())后显示false,无法启用cuda
解决办法,更新cuda,重新下载最新的pytorch后完美解决
检查代码
import torch
from torch.backends import cudnn
print(torch.__version__)
print(torch.cuda.device_count())
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())