diff --git a/Dockerfile b/Dockerfile index 5ef9e83..c15e643 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,5 @@ # Start with a base docker image that contains torch and cutorch. -FROM kaixhin/cuda-torch +FROM kaixhin/cuda-torch:7.5 # Install fblualib and its dependencies : ADD install_all.sh /root/install_all.sh diff --git a/install_all.sh b/install_all.sh index fc9cf78..d42f348 100644 --- a/install_all.sh +++ b/install_all.sh @@ -95,7 +95,8 @@ if [ $current -eq 1 ]; then else git clone -b v0.35.0 --depth 1 https://github.com/facebook/folly git clone -b v0.24.0 --depth 1 https://github.com/facebook/fbthrift - git clone -b v1.0 https://github.com/facebook/thpp +# git clone -b v1.0 https://github.com/facebook/thpp + git clone -b v1+patch-1 https://github.com/dav1nci/thpp git clone -b v1.0 https://github.com/facebook/fblualib fi