Update Habana environment to 1.20.0 and PyTorch 2.6.0

habana/Dockerfile.habana

@@ -1,8 +1,8 @@
-FROM vault.habana.ai/gaudi-docker/1.9.0/ubuntu20.04/habanalabs/tensorflow-installer-tf-cpu-2.11.0:latest
-
+FROM vault.habana.ai/gaudi-docker/1.20.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
 LABEL maintainer="Javier Duarte <jduarte@ucsd.edu>"
 
+RUN apt-get update && \
+    apt-get install -qq -y graphviz graphviz-dev
+
 COPY requirements.txt .
-COPY requirements_nodeps.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-RUN pip install --no-cache-dir -r requirements_nodeps.txt --no-deps
+RUN pip install -r requirements.txt