From ca223b2f00eebbed5741a500d4d09812ad585cd5 Mon Sep 17 00:00:00 2001 From: lltcggie Date: Thu, 18 Oct 2018 00:35:18 +0900 Subject: [PATCH] =?UTF-8?q?CUDA=E3=82=A2=E3=83=BC=E3=82=AD=E3=83=86?= =?UTF-8?q?=E3=82=AF=E3=83=81=E3=83=A3=E3=82=92=E6=8C=87=E5=AE=9A=E3=81=99?= =?UTF-8?q?=E3=82=8B=E3=82=88=E3=81=86=E3=81=AB=E3=81=97=E3=81=9F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- caffe | 2 +- caffe_build.bat | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) 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