Runtimeerror cuda error invalid device ordinal

Traceback (most recent call last):
  File "bert_att_weight_retrieval.py", line 381, in <module>
    main()
  File "bert_att_weight_retrieval.py", line 244, in main
    model.to(device)
  File "/run/media/marco/Daten/GitHub/feature-importance/venv36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 381, in to
    return self._apply(convert)
  File "/run/media/marco/Daten/GitHub/feature-importance/venv36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply
    module._apply(fn)
  File "/run/media/marco/Daten/GitHub/feature-importance/venv36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply
    module._apply(fn)
  File "/run/media/marco/Daten/GitHub/feature-importance/venv36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply
    module._apply(fn)
  File "/run/media/marco/Daten/GitHub/feature-importance/venv36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 193, in _apply
    param.data = fn(param.data)
  File "/run/media/marco/Daten/GitHub/feature-importance/venv36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 379, in convert
    return t.to(device, dtype if t.is_floating_point() else None, non_blocking)
RuntimeError: CUDA error: invalid device ordinal

Collecting environment information...
PyTorch version: 1.0.1.post2
Is debug build: No
CUDA used to build PyTorch: 9.0.176

OS: Manjaro Linux
GCC version: (Arch Linux 9.3.0-1) 9.3.0
CMake version: Could not collect

Python version: 3.6
Is CUDA available: Yes
CUDA runtime version: Could not collect
GPU models and configuration: GPU 0: GeForce GTX 1660
Nvidia driver version: 440.82
cuDNN version: Could not collect

Versions of relevant libraries:
[pip3] msgpack-numpy==0.4.3.2
[pip3] numpy==1.17.1
[pip3] torch==1.0.1.post2
[conda] Could not collect

Wed May 27 13:22:05 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.82       Driver Version: 440.82       CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 1660    Off  | 00000000:01:00.0  On |                  N/A |
|  0%   48C    P8     9W / 120W |    413MiB /  5936MiB |      1%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0       750      G   /usr/lib/Xorg                                275MiB |
|    0      2536      G   ...RubyMine/ch-0/201.6668.156/jbr/bin/java     3MiB |
|    0      5829      G   ...AAAAAAAAAAAACAAAAAAAAAA= --shared-files   125MiB |
|    0     23410      G   ...yCharm-P/ch-0/201.6668.115/jbr/bin/java     3MiB |
+-----------------------------------------------------------------------------+

Решение:
Основная причина этой ошибки — причина номера графического процессора или версия установки CUDA не совпадает

Например, когда в локальной области есть только один графический процессор (GPU: 0), а в программе используется графические процессоры: 1.

Поэтому найдите место для определения устройства в программе следующим образом:

device = th. device ("cuda:1")

Измените «CUDA: 1» на «CUDA: 0», и программа не сообщит об ошибке.

device = th. device ("cuda:0")

[size = xx-large] [color = red] DBAccess SQL Server на основе хранимой процедуры, которую я написал ранее, просто вставьте ее (Oracle и другие методы общего доступа похожи): [/ color] [/ size] [size =…

Двоичное значение добавлено: Первый шаг: два различия или: эквивалент каждому биту, несущий; Шаг 2: два числа и левый Shift One: эквивалентно переносу переноски; Шаг 3: Повторите вышеуказанные шаги до…

Во-первых, ручной круглой трансляции 1. Во-первых, вы должны подготовить несколько спецификаций, одинаковый размер (больше двух ОН), название закона, в порядке, ниже моего расположения изображения и и…

params указывает, что параметры являются переменными Без оформления параметров вы можете передать только массив. После добавления параметров можно передавать не только массив, но и несколько параметро…

Иногда требуется некоторое время для загрузки перед запуском программы. Чтобы пользователь не чувствовал, что программа зависает и зависает, специально добавлен загрузочный интерфейс, чтобы напомнить …

ВSETUP После запроса клиент инициируетPLAY Запрос на запрос к серверу начала передачи аудио- и видеоданных. ВPLAY Когда запрос выполняется, он должен быть выполненSETUP Request, клиентский сеанс устан…