1
0
Fork 0
mirror of synced 2024-09-21 12:02:33 +12:00
waifu2x/lib/portable.lua

17 lines
365 B
Lua
Raw Normal View History

2015-08-05 14:49:38 +12:00
require 'nn'
local function load_cuda()
require 'cunn'
end
if pcall(load_cuda) then
require 'cunn'
else
--[[ TODO: fakecuda does not work.
io.stderr:write("use FakeCUDA; if you have NVIDIA GPU, Please install cutorch and cunn. FakeCuda will be extremely slow.\n")
require 'torch'
require 'nn'
require('fakecuda').init(true)
--]]
end