Caffe-GPU compilation problem: nvcc fatal : Unsupported gpu architecture 'compute_20'

environment:

Ubuntu 16.04 

OpenCV 3.3

Miracles 9.0


compilation problem

[html] view plain copy
  1. NVCC src/caffe/layers/bnll_layer.cu  
  2. nvcc fatal   : Unsupported gpu architecture 'compute_20'  
  3. Makefile:594: recipe for target '.build_release/cuda/src/caffe/layers/bnll_layer.o' failed  
  4. make: *** [.build_release/cuda/src/caffe/layers/bnll_layer.o] Error 1  
  5. make: *** Waiting for unfinished jobs....  


Take a closer look at the CUDA_ARCH setting in Makefile.config is not set as specified:
[html] view plain copy
  1. # CUDA architecture setting: going with all of them.  
  2. # For CUDA <6.0, comment the *_50 through *_61 lines for compatibility.   
  3. # For CUDA <8.0, comment the *_60 and *_61 lines for compatibility.   
  4. # For CUDA >= 9.0, comment the *_20 and *_21 lines for compatibility.  
  5. CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \  
  6.                 -gencode arch=compute_20,code=sm_21 \  
  7.                 -gencode arch=compute_30,code=sm_30 \  
  8.                 -gencode arch=compute_35,code=sm_35 \  
  9.                 -gencode arch=compute_50,code=sm_50 \  
  10.                 -gencode arch=compute_52,code=sm_52 \  
  11.                 -gencode arch=compute_60,code=sm_60 \  
  12.                 -gencode arch=compute_61,code=sm_61 \  
  13.                 -gencode arch=compute_61,code=compute_61  

Because I installed CUDA9.0, I can delete the following two lines
[html] view plain copy
  1. -gencode arch=compute_20,code=sm_20 \  
  2. -gencode arch=compute_20,code=sm_21 \ 

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=324961012&siteId=291194637