I am getting the following while running the command "sudo docker build -t caffe:gpu ./standalone/gpu" , but when I use "cpu" instead of "gpu" it works well and it was successfully installed. I want to use it on gpu actually. I am using CUDA version 8 and it is showing architecture error.
nvcc fatal : Unsupported gpu architecture 'compute_60'
CMake Error at cuda_compile_generated_math_functions.cu.o.cmake:206 (message):
Error generating
/opt/caffe/build/src/caffe/CMakeFiles/cuda_compile.dir/util/./cuda_compile_generated_math_functions.cu.o
make[2]: *** [src/caffe/CMakeFiles/cuda_compile.dir/util/./cuda_compile_generated_math_functions.cu.o] Error 1
make[2]: *** Waiting for unfinished jobs....
nvcc fatal : Unsupported gpu architecture 'compute_60'
CMake Error at cuda_compile_generated_mvn_layer.cu.o.cmake:206 (message):
Error generating
/opt/caffe/build/src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_mvn_layer.cu.o
make[2]: *** [src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_mvn_layer.cu.o] Error 1
nvcc fatal : Unsupported gpu architecture 'compute_60'
CMake Error at cuda_compile_generated_adagrad_solver.cu.o.cmake:206 (message):
Error generating
/opt/caffe/build/src/caffe/CMakeFiles/cuda_compile.dir/solvers/./cuda_compile_generated_adagrad_solver.cu.o
make[2]: *** [src/caffe/CMakeFiles/cuda_compile.dir/solvers/./cuda_compile_generated_adagrad_solver.cu.o] Error 1
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
The command '/bin/sh -c git clone -b ${CLONE_TAG} --depth 1
https://github.com/BVLC/caffe.git . && for req in $(cat python/requirements.txt) pydot; do pip install $req; done && mkdir build && cd build && cmake -DUSE_CUDNN=1 .. && make -j"$(nproc)"' returned a non-zero code: 2