爲caffe構建GPU docker鏡像時出錯：不支持的GPU架構'compute_60'

試圖構建基於caffe docker鏡像的gpu未成功。運行的命令：爲caffe構建GPU docker鏡像時出錯：不支持的GPU架構'compute_60'

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 MBPro具有GT750m NVIDIA芯片組的運行。

注：CPU基於caffe泊塢窗圖像從GitHub庫：https://github.com/BVLC/caffedocker已成功建立了caffe/docker目錄中在我的Mac：

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版本問題。

有沒有人成功構建了gpu Caffe在Docker容器上的版本從mac？

來源

2016-12-18 javadba

在Linux x86_64服務器上運行相同問題 –

在Ubuntu 14.04 nvidia-docker x86_64服務器上出現同樣的問題。 –

在Ubuntu 14.04 nvidia-docker x86_64服務器上出現同樣的問題 – jjdblast

在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

來源

2017-03-15 06:36:00 cerebrou

爲caffe構建GPU docker鏡像時出錯：不支持的GPU架構'compute_60'

回答

相關問題