diff --git a/pytorch/setup.sh b/pytorch/setup.sh index b32234a7..3c8b8295 100755 --- a/pytorch/setup.sh +++ b/pytorch/setup.sh @@ -10,9 +10,11 @@ docker pull nvcr.io/nvidia/${NAME_NGC} echo "---------------------------------------------------" echo "Clone Repos: " echo "---------------------------------------------------" -git clone https://github.com/LambdaLabsML/DeepLearningExamples.git && \ +export TARGET_DIR=DeepLearningExamples +git -C "$TARGET_DIR" pull || git clone https://github.com/LambdaLabsML/${TARGET_DIR}.git "$TARGET_DIR" && \ cd DeepLearningExamples && \ git checkout lambda/benchmark && \ cd .. -git clone https://github.com/lambdal/deeplearning-benchmark.git && \ +export TARGET_DIR=deeplearning-benchmark +git -C "$TARGET_DIR" pull || git clone https://github.com/lambdal/${TARGET_DIR}.git "$TARGET_DIR" && \ cd deeplearning-benchmark/pytorch