multi GPU server
This commit is contained in:
parent
27423e6381
commit
0a683fcb34
|
@ -1,4 +1,11 @@
|
||||||
#!/bin/zsh
|
#!/bin/zsh
|
||||||
# waifu2x daemon script
|
# waifu2x daemon script
|
||||||
|
gpu=1
|
||||||
|
port=8812
|
||||||
|
if [ $# -eq 2 ]; then
|
||||||
|
gpu=$1
|
||||||
|
port=$2
|
||||||
|
fi
|
||||||
source /home/ubuntu/.zshrc
|
source /home/ubuntu/.zshrc
|
||||||
stdbuf -o 0 th web.lua >> ./waifu2x.log 2>&1
|
echo stdbuf -o 0 th web.lua -gpu $gpu -port $port >> ./waifu2x_${port}.log 2>&1
|
||||||
|
stdbuf -o 0 th web.lua -gpu $gpu -port $port >> ./waifu2x_${port}.log 2>&1
|
||||||
|
|
18
web.lua
18
web.lua
|
@ -4,13 +4,21 @@ local uuid = require 'uuid'
|
||||||
local ffi = require 'ffi'
|
local ffi = require 'ffi'
|
||||||
local md5 = require 'md5'
|
local md5 = require 'md5'
|
||||||
require 'pl'
|
require 'pl'
|
||||||
|
|
||||||
torch.setdefaulttensortype('torch.FloatTensor')
|
|
||||||
torch.setnumthreads(4)
|
|
||||||
|
|
||||||
require './lib/portable'
|
require './lib/portable'
|
||||||
require './lib/LeakyReLU'
|
require './lib/LeakyReLU'
|
||||||
|
|
||||||
|
local cmd = torch.CmdLine()
|
||||||
|
cmd:text()
|
||||||
|
cmd:text("waifu2x-api")
|
||||||
|
cmd:text("Options:")
|
||||||
|
cmd:option("-port", 8812, 'listen port')
|
||||||
|
cmd:option("-gpu", 1, 'Device ID')
|
||||||
|
cmd:option("-core", 2, 'number of CPU cores')
|
||||||
|
local opt = cmd:parse(arg)
|
||||||
|
cutorch.setDevice(opt.gpu)
|
||||||
|
torch.setdefaulttensortype('torch.FloatTensor')
|
||||||
|
torch.setnumthreads(opt.core)
|
||||||
|
|
||||||
local iproc = require './lib/iproc'
|
local iproc = require './lib/iproc'
|
||||||
local reconstruct = require './lib/reconstruct'
|
local reconstruct = require './lib/reconstruct'
|
||||||
local image_loader = require './lib/image_loader'
|
local image_loader = require './lib/image_loader'
|
||||||
|
@ -196,5 +204,5 @@ local app = turbo.web.Application:new(
|
||||||
{"^/api$", APIHandler},
|
{"^/api$", APIHandler},
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
app:listen(8812, "0.0.0.0", {max_body_size = CURL_MAX_SIZE})
|
app:listen(opt.port, "0.0.0.0", {max_body_size = CURL_MAX_SIZE})
|
||||||
turbo.ioloop.instance():start()
|
turbo.ioloop.instance():start()
|
||||||
|
|
Loading…
Reference in a new issue