Я просто потратил некоторое время на это. Ответ Томы не завершен. Скажите, что ваша программа test.py
, вы хотите использовать gpu0 для запуска этой программы и не допускать других gpus.
Вы должны написать CUDA_VISIBLE_DEVICES=0 python test.py
Обратите внимание, что это DEVICES
не DEVICE
Я понял, это был в основном драйвер ftdi, но также немного настроек. Я изложил, что я сделал в своем блоге. https://mitchjonesengineering.us/?p=18