RuntimeError: cuda 运行时错误 (30) : ..\aten\src\THC\THCGeneral.cpp:87 处的未知错误答案

【问题标题】：RuntimeError: cuda runtime error (30) : unknown error at ..\aten\src\THC\THCGeneral.cpp:87RuntimeError: cuda 运行时错误 (30) : ..\aten\src\THC\THCGeneral.cpp:87 处的未知错误
【发布时间】：2020-11-01 19:28:26
【问题描述】：

我尝试在 Windows 10 上安装 CUDA 以在我的 GPU (NVIDIA GeForce 710) 中训练神经网络，但在尝试加载初始模型时出现以下错误。

这是我正在运行的代码：

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

device
"""# 1. Create the classifier"""

C = nn.Sequential(Flatten(), nn.Linear(784,200), nn.ReLU(), 
                            nn.Linear(200,100), nn.ReLU(),
                            nn.Linear(100,100), nn.ReLU(),
                            nn.Linear(100,10))

"""Upload the trained model"""

C.load_state_dict(torch.load("C.pt",map_location='cuda'))

这是我得到的错误：

C.load_state_dict(torch.load("C.pt",map_location='cuda'))
Traceback (most recent call last):

  File "<ipython-input-2-358f76f483ed>", line 1, in <module>
    C.load_state_dict(torch.load("C.pt",map_location='cuda'))

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\serialization.py", line 368, in load
    return _load(f, map_location, pickle_module)

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\serialization.py", line 542, in _load
    result = unpickler.load()

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\serialization.py", line 505, in persistent_load
    data_type(size), location)

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\serialization.py", line 385, in restore_location
    return default_restore_location(storage, map_location)

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\serialization.py", line 114, in default_restore_location
    result = fn(storage, location)

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\serialization.py", line 96, in _cuda_deserialize
    return obj.cuda(device)

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\_utils.py", line 68, in _cuda
    with torch.cuda.device(device):

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\cuda\__init__.py", line 229, in __enter__
    _lazy_init()

  File "C:\Users\usuario\anaconda3\lib\site-packages\torch\cuda\__init__.py", line 162, in _lazy_init
    torch._C._cuda_init()

RuntimeError: cuda runtime error (30) : unknown error at ..\aten\src\THC\THCGeneral.cpp:87

我已经安装了 cudNN，这些是我正在使用的版本。

Python 3.7.6

CUDA 8.0

Pytorch 1.0.1

【问题讨论】：

标签： python pytorch gpu

【解决方案1】：

重新启动计算机可能会解决此问题

reference

如果它不能确定你

重新安装最新的 GPU 驱动程序

重启

确保您拥有管理员权限

【讨论】：

【解决方案2】：

无需重启电脑。将这些命令粘贴到命令终端中，错误将被删除。

sudo rmmod nvidia_uvm

sudo rmmod 英伟达

sudo modprobe nvidia

sudo modprobe nvidia_uvm

【讨论】：