portable.lua 352 B

123456789101112131415
  1. local function load_cuda()
  2. require 'cunn'
  3. end
  4. if pcall(load_cuda) then
  5. require 'cunn'
  6. else
  7. --[[ TODO: fakecuda does not work.
  8. io.stderr:write("use FakeCUDA; if you have NVIDIA GPU, Please install cutorch and cunn. FakeCuda will be extremely slow.\n")
  9. require 'torch'
  10. require 'nn'
  11. require('fakecuda').init(true)
  12. --]]
  13. end