nvcc fatal : Unsupported gpu architecture 'compute_20' while cuda 9.1+caffe+openCV 3.4.0 is installed

后端 未结 3 883
青春惊慌失措
青春惊慌失措 2021-02-09 02:59

I have installed CUDA 9.1+cudnn-9.1+opencv 3.4.0+caffe.

When I tried to run make all -j8 in caffe directory, this error occurred:

3条回答
  •  盖世英雄少女心
    2021-02-09 03:24

    This fixed it for me

    cd caffe && mkdir build && cd build && \
       cmake -DUSE_CUDNN=1 -DUSE_NCCL=1 -DCUDA_ARCH_NAME=Manual -DCUDA_ARCH_BIN="50 52 60 61" .. && \
       sudo make -j"$(nproc)"
    

    Just dropped 20 and 30 arch support

提交回复
热议问题