colab中明明显示有cuda,但仍旧出现此错误。
原来是在使用cuda之前要用代码检测一下cuda的存在。如下:
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
如图,运行成功!
colab中明明显示有cuda,但仍旧出现此错误。
原来是在使用cuda之前要用代码检测一下cuda的存在。如下:
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
如图,运行成功!