为caffe构buildGPU docker镜像时出错:不支持的gpu架构“compute_60”
尝试构build基于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
并在2013年底运行的拥有GT750m
nvidia芯片组的GT750m
上运行。
注意:从他们的github回购: https : //github.com/BVLC/caffe docker
基于CPU的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上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