Question

尝试构建基于{em> gpu 的caffe泊坞窗图像失败了。命令运行：

docker build -t caffe:gpu standalone/gpu

它运行了大约20分钟，包括通过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芯片组的MBPro上运行。

注意：来自其github repo：https://github.com/BVLC/caffe caffe的 cpu 基于docker泊坞窗图片已成功构建于{我的mac上的{1}}目录：

caffe/docker

..大约十分钟后..

docker build -t caffe:cpu standalone/cpu

有人提到不同的项目Successfully built bf71c19501e5遇到类似的错误https://github.com/arrayfire/arrayfire/issues/1535。但是，我缺乏对arrayfire命令进行问题排查的具体知识，以尝试复制看似BVLC/caffe/docker与cuda7.5版本问题的内容。

有没有人在mac上的docker容器上成功构建了 gpu 版本的Caffe？

Answer 1

在Makefile.example中，尝试注释掉* _60和* _61行（为了与CUDA＆lt; 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

为caffe构建GPU docker镜像时出错：不支持的gpu架构'compute_60'

1 个答案: