1
0
Fork 0
mirror of synced 2024-05-17 03:12:18 +12:00
This commit is contained in:
yshhuang 2023-02-25 08:33:27 -05:00 committed by GitHub
commit 503fefef21
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

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