Почему я получаю сообщение "pycuda._driver.LogicError: ошибка cuInit: неправильный порядковый номер устройства"?

У меня есть два графических процессора, CUDA и TensorFlow работают правильно, nvidia-smi Результаты

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.24.02              Driver Version: 396.24.02                 |
|-------------------------------+----------------------+----------------------+
| 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 TIT...  Off  | 00000000:01:00.0  On |                  N/A |
| 29%   46C    P8    16W / 250W |    113MiB /  6080MiB |      1%      Default |
+-------------------------------+----------------------+----------------------+
|   1  P104-100            Off  | 00000000:04:00.0 Off |                  N/A |
| 37%   43C    P8     8W / 180W |      0MiB /  4042MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

но у меня есть некоторые проблемы с PyCuda даже в простом случае

from pycuda import driver as drv
drv.init()

Я получил ошибку pycuda._driver.LogicError: cuInit failed: invalid device ordinal, Кто-нибудь может мне помочь - что это за проблема?

0 ответов

Другие вопросы по тегам