macos - 为caffe构建GPU docker镜像时发生错误:不支持的GPU架构 'compute_60'

Question

尝试构建基于gpu的caffe docker镜像失败。该命令运行:

docker build -t caffe:gpu standalone/gpu

它运行了大约二十分钟，包括通过 cython将许多numpy / scikit脚本编译为c代码。但是构建最终在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并在具有 GT750m nvidia芯片组的2013年末MBPro上运行。

注意:他们的github存储库中基于 cpu 的 caffe docker镜像: https://github.com/BVLC/caffe docker已在我的Mac的 caffe/docker目录内成功构建:

docker build -t caffe:cpu standalone/cpu

..大约十分钟后..

Successfully built bf71c19501e5

有人提到另一个项目 arrayfire遇到类似的错误 https://github.com/arrayfire/arrayfire/issues/1535。但是，我缺乏专门的知识来对 BVLC/caffe/docker命令进行故障排除，以尝试复制似乎是 cuda7.5与 cuda8.0版本控制问题的内容。

有谁成功在Mac上的Docker容器上成功构建了Caffe的 gpu 版本的Caffe？

Answer 1

在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