CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

Update install_lua_modules.sh

Browse source 
Install packages from torch's rocks repo.
graphicsmagick and csvigo are only available on torch's rocks repo.
This commit is contained in:
黄永生 2020-04-23 19:06:57 +08:00 committed by GitHub
parent ed93894fc8
commit 8e1895b435
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
install_lua_modules.sh
View file

@ -1,12 +1,12 @@
#!/bin/bash
CUDNN_BRANCH=R7 # for cudnn7
CUDNN_BRANCH=master # 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
git clone https://github.com/torch/rocks.git -b $CUDNN_BRANCH $CUDNN_WORK_DIR
cd $CUDNN_WORK_DIR
luarocks make cudnn-scm-1.rockspec
cd ..