From 204992737501d0b67b3b8441551bafc4c7a4989e Mon Sep 17 00:00:00 2001 From: jminock Date: Mon, 7 Oct 2024 11:39:27 -0400 Subject: [PATCH] Adding torch Python package to container (#300) Co-authored-by: James Minock --- GetToolDAQ.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/GetToolDAQ.sh b/GetToolDAQ.sh index 543fcff50..7bd2292b3 100644 --- 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