在将 GPU 用于 PyTorch 模型时，出现 CUDA 错误：未知错误？答案

【问题标题】：While using GPU for PyTorch models, getting the CUDA error: Unknown error?在将 GPU 用于 PyTorch 模型时，出现 CUDA 错误：未知错误？
【发布时间】：2019-07-19 13:08:00
【问题描述】：

我正在尝试使用使用 PyTorch 的预训练模型。将模型加载到 GPU 时，出现以下错误：

Traceback (most recent call last):
  File "model\vgg_model.py", line 45, in <module>
    vgg_model1 = VGGFeatureExtractor(True).double().to(device)
  File "C:\Users\myidi\Anaconda3\envs\openpose\lib\site-packages\torch\nn\modules\module.py", line 386, in to
    return self._apply(convert)
  File "C:\Users\myidi\Anaconda3\envs\openpose\lib\site-packages\torch\nn\modules\module.py", line 193, in _apply
    module._apply(fn)
  File "C:\Users\myidi\Anaconda3\envs\openpose\lib\site-packages\torch\nn\modules\module.py", line 193, in _apply
    module._apply(fn)
  File "C:\Users\myidi\Anaconda3\envs\openpose\lib\site-packages\torch\nn\modules\module.py", line 199, in _apply
    param.data = fn(param.data)
  File "C:\Users\myidi\Anaconda3\envs\openpose\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 "C:\Users\myidi\Anaconda3\envs\openpose\lib\site-packages\torch\cuda\__init__.py", line 163, in _lazy_init
    torch._C._cuda_init()
RuntimeError: CUDA error: unknown error

我有一台 Windows 10 笔记本电脑、Nvidia 940m GPU、最新的 Pytorch 和 CUDA Toolkit 9.0（也在 10.0 上尝试过）。

我尝试重新安装 GPU 驱动程序，重新启动我的机器，重新安装 PyTorch、Torchvision 和 CUDA Toolkit。

同时使用以下命令查看 PyTorch 是否正在检测 GPU：

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

我得到以下输出：device(type='cuda')。

可能存在哪些问题？我已经尝试过这里提到的解决方案：https://github.com/pytorch/pytorch/issues/20990，但问题仍然存在。

我只是将torch.cuda.current_device() 放在import torch 之后，但问题仍然存在。

【问题讨论】：

标签： pytorch

【解决方案1】：

奇怪的是，这可以通过使用 CUDA Toolkit 10.1 来实现。我不知道为什么最新的不是 PyTorch 网站上他们提供下载库的命令的部分中的默认版本。

使用以下命令安装库：conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

【讨论】：