Adding torch Python package to container (#300)

Co-authored-by: James Minock <jminock@anniegpvm02.fnal.gov>
ANNIEsoft · Oct 7, 2024 · 2049927 · 2049927
1 parent ca22bc5
commit 2049927
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/GetToolDAQ.sh b/GetToolDAQ.sh
@@ -497,6 +497,7 @@ then
     pip3 install uproot==4.3.7
     pip3 install xgboost==1.6.2
     pip3 install tensorflow==2.10.0
+    pip3 install torch==1.12.1
     pip3 install PyQt5
     # set tensorflow verbosity to suppress info messages about not having a GPU or maximal acceleration
     # https://stackoverflow.com/questions/35911252/disable-tensorflow-debugging-information/42121886#42121886