Преглед на файлове

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