Question

我正在尝试通过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之后，但问题仍然存在。

Answer 1

奇怪的是，这是通过使用CUDA Toolkit 10.1进行的。我不知道为什么最新版本不是PyTorch网站上默认的版本，因为它们提供了下载库的命令。

使用以下命令安装库：import pandas as pd model_1 = pd.DataFrame(data={"observation_id": [0, 1, ..., 1000], "param_1": [0, 4, ..., 2]}) model_2 = pd.DataFrame(data={"observation_id": [0, 2, ..., 1000], "param_1": [1, 3, ..., 1]})

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

1 个答案: