Added GPUtil to DVAP's dockerfile

TsotsosLab · Nov 11, 2022 · 19cf041 · 19cf041
1 parent 5f131b7
commit 19cf041
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dockerfiles/Dockerfile.DVAP b/dockerfiles/Dockerfile.DVAP
@@ -34,6 +34,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
  git && \
  rm -rf /var/lib/apt/lists/*
 
+################################################################################
+### gputil for CPU/GPU toggling
+RUN pip install gputil
+
 ################################################################################
 ### @wenguanwang's implementation depends on HED's caffe.