浏览代码

Add install_lua_modules.sh

nagadomi 7 年之前
父节点
当前提交
9f4626df80
共有 1 个文件被更改,包括 21 次插入0 次删除
  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