hughperkins / pytorch

Python wrappers for torch and lua

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is LuaJIT supported as well?

abhitopia opened this issue · comments

When I run my lua code using PyTorch, it seems to run Lua5.1 and not LuaJIT, any way I can change that?

So, what should happen is:

  • on linux, it will use luajit
  • on mac it will use lua

Can you confirm which platform you are using?

By the way, if you're on Mac, and wondering "why use lua on Mac", there are some complicated technical reasons involving memory: http://luajit.org/install.html

If you're building a 64 bit application on OSX which links directly or indirectly against LuaJIT, you need to link your main executable with these flags:

-pagezero_size 10000 -image_base 100000000

Also, it's recommended to rebase all (self-compiled) shared libraries which are loaded at runtime on OSX/x64 (e.g. C extension modules for Lua). See: man rebase

Since our executable is python, we would need to rebuild python to do this. If you do want to rebuild python, to include these flags, we can probably work together to get pytorch to detect your version of python, or have an appropriate switch, so that it will link with luajit, instead of lua, on your system.

@hughperkins - You are right, I was indeed using my mac for the purpose. I am exploring TensorFlow now, and probably only come back to torch if it doesn't work for me. Thanks though.

ok, fair enough :-)