CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-13 08:24:30 +12:00
Code Issues Releases Wiki Activity

Merge 8e1895b435 into cc385f97a9

Browse source
This commit is contained in:
yshhuang 2023-02-25 08:33:27 -05:00 committed by GitHub
parent cc385f97a9 8e1895b435
commit 503fefef21
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 #!/bin/bash
CUDNN_BRANCH=R7 # for cudnn7 CUDNN_BRANCH=master # for cudnn7
CUDNN_WORK_DIR=.cudnn CUDNN_WORK_DIR=.cudnn
install_cudnn() install_cudnn()
{ {
rm -fr $CUDNN_WORK_DIR 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 cd $CUDNN_WORK_DIR
luarocks make cudnn-scm-1.rockspec luarocks make cudnn-scm-1.rockspec
cd .. cd ..