cuda报错, RuntimeError: CUDA error: unknown error

转自：Pytorch 常见问题
苦恼了一下午，再次感谢作者！

cuda报错, RuntimeError: CUDA error: unknown error

  File "D:\ProgramFiles\ProgramFiles\anaconda\lib\site-packages\torch\nn\modules\module.py", line 384, in convert
    return t.to(device, dtype if t.is_floating_point() else None, non_blocking)
  File "D:\ProgramFiles\ProgramFiles\anaconda\lib\site-packages\torch\cuda\__init__.py", line 163, in _lazy_init
    torch._C._cuda_init()
RuntimeError: CUDA error: unknown error

在import后边加上下边一句：torch.cuda.current_device()

import torch
torch.cuda.current_device()
import torchvision