Browse Source

Add install_lua_modules.sh

nagadomi 7 years ago
parent
commit
9f4626df80
1 changed files with 21 additions and 0 deletions
  1. 21 0
      install_lua_modules.sh

+ 21 - 0
install_lua_modules.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+CUDNN_BRANCH=R7 # for cudnn7
+CUDNN_WORK_DIR=.cudnn
+
+install_cudnn()
+{
+    rm -fr $CUDNN_WORK_DIR
+    git clone https://github.com/soumith/cudnn.torch.git -b $CUDNN_BRANCH $CUDNN_WORK_DIR
+    cd $CUDNN_WORK_DIR
+    luarocks make cudnn-scm-1.rockspec
+    cd ..
+    rm -fr $CUDNN_WORK_DIR
+}
+
+luarocks install graphicsmagick
+luarocks install lua-csnappy
+luarocks install md5
+luarocks install csvigo
+install_cudnn
+PREFIX=$HOME/torch/install luarocks install turbo