запуск ffmpeg с nvenc внутри докера nvidia

Question

Я установил Nvidia Video Codec SDK 8.2 + ffmpeg внутри контейнера докера с помощью nvidia-docker , но когда я запускаю этот

ffmpeg -f rawvideo -s:v 1920x1080 -r 30 -pix_fmt yuv420p -i HeavyHand_1080p.yuv -c:v h264_nvenc -preset slow -cq 10 -bf 2 -g 150 output.mp4

Я получил это ошибка

Невозможно загрузить libnvidia-encode.so.1

Минимальный требуемый драйвер Nvidia для nvenc - 390,25 или новее Ошибка инициализации выходного потока 0: 0 - Ошибка при открытии датчика для выходного потока # 0: 0 - возможно, неверные параметры, такие как bit_rate, скорость, ширина или высота

В противном случае nvidia-smi отображает это

используется этот графический процессор GeForce 1050 Ti и версия cuda 9.0

Answer 1

Если вы используете nvidia-docker для запуска образа docker, то все должно быть взято с хоста. Сначала вы должны установить cuda на хост-систему. "sudo apt-get install cuda" и затем запустите образ докера.

Вы также должны использовать образ nvidia-cuda вместо linux (в файле docker)