cuda - RuntimeError：CUDA错误（10）：设备序号无效

Traceback (most recent call last):
  File "main.py", line 29, in <module>
    p = Processor(sys.argv[2:])
  File "D:\tools\workspace\st-gcn\processor\processor.py", line 29, in __init__
    self.init_environment()
  File "D:\tools\workspace\st-gcn\processor\processor.py", line 38, in init_environment
    super().init_environment()
  File "D:\tools\workspace\st-gcn\processor\io.py", line 62, in init_environment
    torchlight.occupy_gpu(gpus)
  File "D:\tools\Anaconda\lib\site-packages\torchlight-1.0-py3.6.egg\torchlight\gpu.py", line 35, in occupy_gpu
RuntimeError: CUDA error (10): invalid device ordinal

我尝试训练模型时发生错误，也许这与如何更改pytorch gpu设置有关？

RuntimeError：CUDA错误（10）：设备序号无效

0 个答案: