Error when Building GPU docker image for caffe: Unsupported gpu architecture 'compute_60'

时间:2023-03-10 01:39:32
Error when Building GPU docker image for caffe: Unsupported gpu architecture 'compute_60'

issue:

Error when Building GPU docker image for caffe: Unsupported gpu architecture 'compute_60'

reason:

CUDA < 8.0

solution:

In the Makefile.example, try commenting out the *_60 and *_61 lines (for compatibility with 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