커피는 CMakeLists를 통과합니다.txt 사용자 정의 GPU ARCH용 동적 라이브러리 컴파일

423 단어 필기DL
1. 카페 디렉토리 아래 cmake 폴더 아래 Cuda를 수정합니다.cmake 파일, 다음 행을 수정합니다.
set(Caffe_known_gpu_archs "30 35 50 52 61")

2. 컴파일할 때 컴파일 옵션을 설정합니다.
cmake -DCUDA_ARCH_NAME=All ..

좋은 웹페이지 즐겨찾기