Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 21:52:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

CUDAアーキテクチャを指定するようにした

Browse Source
This commit is contained in:
lltcggie 2018-10-18 00:35:18 +09:00
parent ccaaf677f7
commit ca223b2f00
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
caffe

@ -1 +1 @@
Subproject commit afb248ff8692c0b2ecaf2a37e138345393a32f64 Subproject commit 599810fc029c6086da112e6cb478fa17439e4594

6
caffe_build.bat
View File

@ -1,5 +1,7 @@
@echo off @echo off
cd /d "%~dp0"
cd caffe cd caffe
set MSVC_VERSION=14 set MSVC_VERSION=14
@ -11,7 +13,7 @@ set USE_LMDB=0
set USE_HDF5=0 set USE_HDF5=0
set WITH_NINJA=0 set WITH_NINJA=0
set CPU_ONLY=0 set CPU_ONLY=0
set CUDA_ARCH_NAME=Auto set CUDA_ARCH_NAME=Manual
set CMAKE_CONFIG=Release set CMAKE_CONFIG=Release
set WITH_CUDA=1 set WITH_CUDA=1
set BUILD_PYTHON=0 set BUILD_PYTHON=0
@ -19,5 +21,7 @@ set BUILD_PYTHON_LAYER=0
set RUN_TESTS=0 set RUN_TESTS=0
set CMAKE_INSTALL_PREFIX=../../lib set CMAKE_INSTALL_PREFIX=../../lib
set RUN_INSTALL=1 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 scripts\build_win.cmd