mirror of
https://github.com/lltcggie/waifu2x-caffe.git
synced 2025-11-24 18:44:55 +00:00
28 lines
474 B
Batchfile
28 lines
474 B
Batchfile
@echo off
|
|
|
|
cd /d "%~dp0"
|
|
|
|
cd caffe
|
|
|
|
set MSVC_VERSION=17
|
|
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
|
|
set CUDA_ARCH_NAME=Manual
|
|
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
|
|
rem set CUDA_ARCH_BIN=75 80 86 89 90 100 120
|
|
set CUDA_ARCH_BIN=89
|
|
set CUDA_ARCH_PTX=120
|
|
|
|
scripts\build_win.cmd |