video-caffe 및 C3D compute_60 error 문제 해결 방법

2720 단어 caffe
GitHub에서 설명한 대로 설치합니다. 단계는 다음과 같습니다.
  • git clone [email protected]:chuckcho/video-caffe.git
  • cd video-caffe
  • mkdir build && cd build
  • cmake .. Make sure CUDA and CuDNN are detected and their paths are correct.
  • make all
  • make install
  • (optional) make runtest

  • 내 그래픽은pascal 모드를 지원하지 않습니다.makefile을 수정했지만.config 하지만 여전히 compute_60의 오류입니다.그래서 카페/cmake/Cuda를 수정했습니다.cmake장Caffe_known_gpu_archs 뒤에 6061 빼기.
    #set(Caffe_known_gpu_archs "20 21(20) 30 35 50 60 61")
    set(Caffe_known_gpu_archs "20 21(20) 30 35 50")

    Pascal 부분을 같이 주석을 달았어요.
     if(${CUDA_ARCH_NAME} STREQUAL "Fermi")
        set(__cuda_arch_bin "20 21(20)")
      elseif(${CUDA_ARCH_NAME} STREQUAL "Kepler")
        set(__cuda_arch_bin "30 35")
      elseif(${CUDA_ARCH_NAME} STREQUAL "Maxwell")
        set(__cuda_arch_bin "50")
     # elseif(${CUDA_ARCH_NAME} STREQUAL "Pascal")
     #   set(__cuda_arch_bin "60 61")
      elseif(${CUDA_ARCH_NAME} STREQUAL "All")
        set(__cuda_arch_bin ${Caffe_known_gpu_archs})
      elseif(${CUDA_ARCH_NAME} STREQUAL "Auto")
        caffe_detect_installed_gpus(__cuda_arch_bin)
      else()  # (${CUDA_ARCH_NAME} STREQUAL "Manual")
        set(__cuda_arch_bin ${CUDA_ARCH_BIN})
      endif()
    

    이 두 곳의 수정을 끝내고 실행에 성공했습니다.

    좋은 웹페이지 즐겨찾기