Привет всем. Как при сборке контейнера передать в него device?(в

Question

Привет всем. Как при сборке контейнера передать в него device?(в

частности GPU) Сразу отвечу на вопросы а-ля "а зачем тебе это надо". Тот же coffe при сборке проверяет какие архитектуры исполнения доступны для ускорителя и соответсвенно собирает ядра под все доступные. а без этого он начинает собирать воообще под все и для 10й куды падает, потому что 20я и 21я архитектуры почили.

#backend #devops #docker #programming #russian

0

18.09.2019

3 ответов

13 просмотров

George Gaál

runtime:nvidia environment: - CUDA_VISIBLE_DEVICE=1

0

18.09.2019

Аббисинские Псы Автор вопроса

#вмемориз сам с собою в тихоря.... В общем гулёж и просовывание целиком /dev - тоже не помогло. на будущее будете собирать кофе в докере - надо менять cmake/Cuda.cmake примерно так RUN cat /src/caffe/cmake/Cuda.cmake | sed -e 's/set(Caffe_known_gpu_archs\ \"20\ 21(20)\ 30\ 35\ 50\ 60\ 61\")/set(Caffe_known_gpu_archs\ \"30\ 35\ 50\ 60\ 61\")/g' > /tmp/1 && \ cp /tmp/1 /src/caffe/cmake/Cuda.cmake

0

18.09.2019

kSandr · Accepted Answer

kSandr

а типа просто параметром сконфигурить билд нельзя чтоль ? ... лол

0

18.09.2019

172 похожих чатов

Привет всем. Как при сборке контейнера передать в него device?(в

3 ответов

Похожие вопросы