diff --git a/caffe b/caffe index afb248f..599810f 160000 --- a/caffe +++ b/caffe @@ -1 +1 @@ -Subproject commit afb248ff8692c0b2ecaf2a37e138345393a32f64 +Subproject commit 599810fc029c6086da112e6cb478fa17439e4594 diff --git a/caffe_build.bat b/caffe_build.bat index 720ac54..6529da3 100644 --- a/caffe_build.bat +++ b/caffe_build.bat @@ -1,5 +1,7 @@ @echo off +cd /d "%~dp0" + cd caffe set MSVC_VERSION=14 @@ -11,7 +13,7 @@ set USE_LMDB=0 set USE_HDF5=0 set WITH_NINJA=0 set CPU_ONLY=0 -set CUDA_ARCH_NAME=Auto +set CUDA_ARCH_NAME=Manual set CMAKE_CONFIG=Release set WITH_CUDA=1 set BUILD_PYTHON=0 @@ -19,5 +21,7 @@ set BUILD_PYTHON_LAYER=0 set RUN_TESTS=0 set CMAKE_INSTALL_PREFIX=../../lib set RUN_INSTALL=1 +set CUDA_ARCH_BIN=30 35 50 52 53 60 61 62 70 72 +set CUDA_ARCH_PTX=72 scripts\build_win.cmd \ No newline at end of file