Создание образа докера с библиотеками cuda на компьютере с Windows - PullRequest
0 голосов
/ 10 апреля 2019

В настоящее время я внедряю решение для машинного обучения, использующее тензорный поток, и внедряю его с использованием колбы и Кубернетеса для масштабирования. В настоящее время, поскольку мой персональный компьютер недостаточно мощный, я использую ноутбук друга с CUDA, CUDnn, чтобы проверить мой ML с GPU nvidia. Сервер реагирует, как и ожидалось, при локальном запуске на машине (с использованием gpu), и он отлично работает при создании образа докера с процессором tenorflow.

Чтобы сделать это немного быстрее, я собираюсь использовать kubernetes с gpu и создать образ докера с общими библиотеками CUDA. Но согласно их документации, я не могу проверить это с Windows, так как nvidia-docker не доступен для Windows. Поэтому я слепо строю образ, используя nvidia / cuda: 9.0-runtime release в качестве основы, устанавливая все необходимые вещи, как и мои 2 выше изображения.

Это все сработало, но в настоящее время я получаю сообщение об ошибке

libcuda.so.1: cannot open shared object file: No such file or directory

Справедливо ли предположить, что эта ошибка появляется, поскольку я не использую машину с Ubuntu с установленными драйверами Nvidia? И можно ли предположить, что мой образ будет работать на движке Kubernetes?

p.s Я не знаю, предназначен ли этот вопрос для обсуждения здесь. Я также отправил это в суперпользователя. Любое руководство по этому вопросу приветствуется

...