Skip to content

Commit

Permalink
Change docstring of setup_libtorch.sh to options which lead to valid …
Browse files Browse the repository at this point in the history
…download directories. #closes idaholab#29024
  • Loading branch information
ke7kto committed Nov 7, 2024
1 parent 681f13d commit b8feaf4
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion scripts/setup_libtorch.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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."
Expand Down

0 comments on commit b8feaf4

Please sign in to comment.