错误类型
import torch
torch.cuda.is_available()
返回True
a = torch.tensor([1])
a.cuda()
But but but 出现 RuntimeError: CUDA error: unknown error
版本
linux: cuda 10.0 cudnn 7.6
pytorch: 1.4.0
原因&解决方法
原因: pytorch 1.4.0 不支持 cuda 10.0
方法: 将pytorch版本降到1.3.0 (如下)
conda install pytorch==1.3.0 torchvision cudatoolkit=10.0 -c pytorch