为caffe构buildGPU docker镜像时出错:不支持的gpu架构“compute_60”

尝试构build基于gpucaffe 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并在2013年底运行的拥有GT750m nvidia芯片组的GT750m上运行。

注意:从他们的github回购: https : //github.com/BVLC/caffe docker基于CPUcaffe 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.5cuda8.0版本问题。

有没有人成功地在Mac上build立了docker容器上的Caffe gpu版本?

在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