portable.lua 365 B

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