Python RuntimeError:cuda运行时错误（30）：位于..\aten\src\THC\THCGeneral.cpp:87的未知错误

我曾尝试在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

C.load\u state\u dict（火炬装载（“C.pt”，map\u location='cuda'））
回溯（最近一次呼叫最后一次）：
文件“”，第1行，在
C.load\u state\u dict（火炬装载（“C.pt”，地图位置='cuda'））
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\serialization.py”，第368行，已加载
返回加载（f，映射位置，pickle模块）
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\serialization.py”，第542行，加载
结果=unpickler.load（）
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\serialization.py”，第505行，持续加载
数据类型（大小、位置）
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\serialization.py”，第385行，位于还原位置
返回默认位置（存储、映射位置）
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\serialization.py”，第114行，默认位置为\u restore\u
结果=fn（存储、位置）
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\serialization.py”，第96行，反序列化
返回对象cuda（设备）
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\\ utils.py”，第68行，在cuda中
带torch.cuda.device（设备）：
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\cuda\\ uuuu init\uuuu.py”，第229行，输入__
_lazy_init（）
文件“C:\Users\usuario\anaconda3\lib\site packages\torch\cuda\\ uuuu init\uuuu.py”，第162行，在\u lazy\u init中
火炬._C._cuda_init（）
RuntimeError:cuda运行时错误（30）：位于..\aten\src\THC\THCGeneral.cpp:87的未知错误

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

Python 3.7.6

CUDA 8.0

---

Pytorch 1.0.1

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

如果不能确定你

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

重新启动

---

确保您具有管理员权限

无需重新启动电脑。在命令终端中突出粘贴这些命令，错误将被删除

sudo rmmod nvidia_uvm

sudo rmmod nvidia

苏多莫迪亚酒店

sudo modprobe nvidia_uvm