Ошибка при создании образа докера графического процессора для кафе: неподдерживаемая архитектура графического процессора 'compute_60'

Question 1

Попытка создать образ докера на основе gpu caffe не удалась. Команда запускается:

docker build -t caffe:gpu standalone/gpu

Он работал около двадцати минут, включая компиляцию многих сценариев numpy / scikit в c-код через cython. Однако сборка окончательно завершилась неудачно на шаге nvidia nvcc:

[  1%] [  1%] Building NVCC (Device) object src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_cudnn_pooling_layer.cu.o
Building NVCC (Device) object src/caffe/CMakeFiles/cuda_compile.dir/util/./cuda_compile_generated_im2col.cu.o
nvcc fatal   : Unsupported gpu architecture 'compute_60'
CMake Error at cuda_compile_generated_cudnn_pooling_layer.cu.o.cmake:206 (message):
  Error generating
  /opt/caffe/build/src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_cudnn_pooling_layer.cu.o


make[2]: *** [src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_cudnn_pooling_layer.cu.o] Error 1
make[2]: *** Waiting for unfinished jobs....
nvcc fatal   : Unsupported gpu architecture 'compute_60'
CMake Error at cuda_compile_generated_im2col.cu.o.cmake:206 (message):
  Error generating

/opt/caffe/build/src/caffe/CMakeFiles/cuda_compile.dir/util/./cuda_compile_generated_im2col.cu.o


make[2]: *** [src/caffe/CMakeFiles/cuda_compile.dir/util/./cuda_compile_generated_im2col.cu.o] Error 1
make[1]: *** [src/caffe/CMakeFiles/caffe.dir/all] Error 2
make: *** [all] Error 2

Я использую El Capitan и использую MBPro конца 2013 года с чипсетом GT750m nvidia.

Примечание. Образ докера на основе cpu caffe из их репозитория на github: https://github.com/BVLC/caffe docker был успешно создан в каталоге caffe/docker на моем Mac:

docker build -t caffe:cpu standalone/cpu

.. и примерно через десять минут ..

Successfully built bf71c19501e5

Есть упоминания о другом проекте arrayfire, в котором возникла аналогичная ошибка https: // github. ru / arrayfire / arrayfire / issues / 1535. Однако мне не хватает специальных знаний для устранения неполадок с командами BVLC/caffe/docker, чтобы попытаться воспроизвести то, что, по-видимому, связано с проблемами управления версиями cuda7.5 и cuda8.0.

Кто-нибудь успешно построил версию Caffe gpu в контейнере докеров с Mac?

macos docker caffe

2

WestCoastProjects 18 Дек 2016 в 23:48

Столкнулся с той же проблемой на сервере linux x86_64

–

jeremy_rutman

9 Янв 2017 в 20:19

Такая же проблема на сервере Ubuntu 14.04 nvidia-docker x86_64.

–

Jeff Benshetler

15 Янв 2017 в 01:49

Такая же проблема на сервере Ubuntu 14.04 nvidia-docker x86_64

–

jjdblast

7 Мар 2017 в 20:13

Answer

В примере Makefile.example попробуйте закомментировать строки * _60 и * _61 (для совместимости с CUDA <8.0).

CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
        -gencode arch=compute_20,code=sm_21 \
        -gencode arch=compute_30,code=sm_30 \
        -gencode arch=compute_35,code=sm_35 \
        -gencode arch=compute_50,code=sm_50 \
        -gencode arch=compute_52,code=sm_52 \
        #-gencode arch=compute_60,code=sm_60 \
        #-gencode arch=compute_61,code=sm_61 \
        #-gencode arch=compute_61,code=compute_61

2

cerebrou 15 Мар 2017 в 09:36

Question 2

В примере Makefile.example попробуйте закомментировать строки * _60 и * _61 (для совместимости с CUDA <8.0).

CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
        -gencode arch=compute_20,code=sm_21 \
        -gencode arch=compute_30,code=sm_30 \
        -gencode arch=compute_35,code=sm_35 \
        -gencode arch=compute_50,code=sm_50 \
        -gencode arch=compute_52,code=sm_52 \
        #-gencode arch=compute_60,code=sm_60 \
        #-gencode arch=compute_61,code=sm_61 \
        #-gencode arch=compute_61,code=compute_61

2

cerebrou 15 Мар 2017 в 09:36

Ошибка при создании образа докера графического процессора для кафе: неподдерживаемая архитектура графического процессора 'compute_60'

1 ответ

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

Новые вопросы

macos