From b8feaf41e3057da294b4499d3ce6972da1705ced Mon Sep 17 00:00:00 2001 From: "Riet, Adriaan Anthony" Date: Thu, 7 Nov 2024 12:38:59 -0700 Subject: [PATCH 1/2] Change docstring of setup_libtorch.sh to options which lead to valid download directories. #closes #29024 --- scripts/setup_libtorch.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/setup_libtorch.sh b/scripts/setup_libtorch.sh index 33771e94fd41..032f7c5d8284 100755 --- a/scripts/setup_libtorch.sh +++ b/scripts/setup_libtorch.sh @@ -48,7 +48,7 @@ if [[ -n "$HELP" ]]; then echo "--cleanup Remove the downloaded tarball after the install" echo "--version=VERSION_NUMBER Specify the version number of libtorch" echo "--libtorch-dest=DESTINATION Specify where the packages are to be copied" - echo "--libtorch-distribution=DISTRIBUTION Specify the distribution (cpu/cuda)" + echo "--libtorch-distribution=DISTRIBUTION Specify the distribution (cpu/cu118/cu121/cu124/rocm6.2)" echo "*************************************************************************************" echo "" exit 0 @@ -173,6 +173,7 @@ PACKAGE=${TORCH_DEST}/${FILENAME} if [[ -f $PACKAGE ]]; then echo "Found requested package for libtorch v. $VERSION, no need to download." else + echo https://download.pytorch.org/libtorch/$TORCH_DISTRIBUTION/$FILENAME curl -L $IGNORE_CERT -o $PACKAGE https://download.pytorch.org/libtorch/$TORCH_DISTRIBUTION/$FILENAME fi echo "Extracting $PACKAGE." From 9c4ad516355907bfcda2cef713f6a0e7ec8584da Mon Sep 17 00:00:00 2001 From: Adriaan Riet Date: Thu, 7 Nov 2024 12:44:42 -0700 Subject: [PATCH 2/2] Update setup_libtorch.sh Remove line used in debugging --- scripts/setup_libtorch.sh | 1 - 1 file changed, 1 deletion(-) diff --git a/scripts/setup_libtorch.sh b/scripts/setup_libtorch.sh index 032f7c5d8284..6fe725e2c6f5 100755 --- a/scripts/setup_libtorch.sh +++ b/scripts/setup_libtorch.sh @@ -173,7 +173,6 @@ PACKAGE=${TORCH_DEST}/${FILENAME} if [[ -f $PACKAGE ]]; then echo "Found requested package for libtorch v. $VERSION, no need to download." else - echo https://download.pytorch.org/libtorch/$TORCH_DISTRIBUTION/$FILENAME curl -L $IGNORE_CERT -o $PACKAGE https://download.pytorch.org/libtorch/$TORCH_DISTRIBUTION/$FILENAME fi echo "Extracting $PACKAGE."