1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00

Add install_lua_modules.sh

This commit is contained in:
nagadomi 2018-06-05 16:00:55 +09:00
parent 929c7f85a9
commit 9f4626df80

21
install_lua_modules.sh Executable file
View file

@ -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