portable.lua 242 B

1234567891011121314151617
  1. require 'torch'
  2. require 'nn'
  3. local function load_cuda()
  4. require 'cutorch'
  5. require 'cunn'
  6. end
  7. local function load_cudnn()
  8. require 'cudnn'
  9. --cudnn.fastest = true
  10. end
  11. if pcall(load_cuda) then
  12. else
  13. end
  14. if pcall(load_cudnn) then
  15. end