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
|
|
|
|
|
|
|
|
|
|
set MSVC_VERSION=14
|
|
|
|
|
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
|
2020-06-28 01:41:37 +09:00
|
|
|
set CUDA_ARCH_BIN=30 35 50 52 53 60 61 62 70 75
|
|
|
|
|
set CUDA_ARCH_PTX=75
|
2018-10-17 03:54:13 +09:00
|
|
|
|
|
|
|
|
scripts\build_win.cmd
|
