Merge pull request #29025 from ke7kto/libtorch_setup_helpstring

Change docstring of setup_libtorch.sh
idaholab · Dec 5, 2024 · 82cdfd9 · 82cdfd9
2 parents 42cb95a + 9c4ad51
commit 82cdfd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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