2018-10-17 03:54:13 +09:00
|
|
|
@echo off
|
|
|
|
|
|
2018-10-18 00:35:18 +09:00
|
|
|
cd /d "%~dp0"
|
|
|
|
|
|
2018-10-17 03:54:13 +09:00
|
|
|
cd caffe
|
|
|
|
|
|
2025-09-07 21:28:26 +09:00
|
|
|
set MSVC_VERSION=17
|
2018-10-17 03:54:13 +09:00
|
|
|
set USE_CUDNN=1
|
|
|
|
|
set USE_NCCL=0
|
|
|
|
|
set USE_OPENCV=0
|
|
|
|
|
set USE_LEVELDB=0
|
|
|
|
|
set USE_LMDB=0
|
|
|
|
|
set USE_HDF5=0
|
|
|
|
|
set WITH_NINJA=0
|
|
|
|
|
set CPU_ONLY=0
|
2018-10-18 00:35:18 +09:00
|
|
|
set CUDA_ARCH_NAME=Manual
|
2018-10-17 03:54:13 +09:00
|
|
|
set CMAKE_CONFIG=Release
|
|
|
|
|
set WITH_CUDA=1
|
|
|
|
|
set BUILD_PYTHON=0
|
|
|
|
|
set BUILD_PYTHON_LAYER=0
|
|
|
|
|
set RUN_TESTS=0
|
|
|
|
|
set CMAKE_INSTALL_PREFIX=../../lib
|
|
|
|
|
set RUN_INSTALL=1
|
2025-09-07 21:28:26 +09:00
|
|
|
rem set CUDA_ARCH_BIN=75 80 86 89 90 100 120
|
|
|
|
|
set CUDA_ARCH_BIN=89
|
|
|
|
|
set CUDA_ARCH_PTX=120
|
2018-10-17 03:54:13 +09:00
|
|
|
|
|
|
|
|
scripts\build_win.cmd
|
