Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 13:42:48 +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
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