diff --git a/samples/torch/downloadTorch.sh b/samples/torch/downloadTorch.sh index 268e6dc8c12..a83c906ceeb 100755 --- a/samples/torch/downloadTorch.sh +++ b/samples/torch/downloadTorch.sh @@ -6,6 +6,8 @@ NO_CUDA=true # set to false for GPU support if [ ! -d $TH_TARGET_DIRECTORY/include/THNN ]; then git clone https://github.com/pytorch/pytorch.git + # Current pytorch master fails the build so we need to checkout a correct revision. + cd pytorch && git checkout 310c3735b9eb97f30cee743b773e5bb054989edc^ && cd ../ mkdir build_torch cd build_torch