From a49838ceb37464139a799e96856ddc45f93751a2 Mon Sep 17 00:00:00 2001
From: PiotrBLL <piotr.sobieszczyk@bluelabellabs.com>
Date: Wed, 6 Nov 2024 23:44:42 +0100
Subject: [PATCH] Add Pytorch Docker images info in README.md

---
 README.md | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/README.md b/README.md
index ddbdd2d..fe1b603 100644
--- a/README.md
+++ b/README.md
@@ -148,6 +148,9 @@ print(result.text)
 docker build -t whisper_hpu:latest -f Dockerfile_hpu .
 ```
 
+In the `Dockerfile_hpu`, we use the `vault.habana.ai/gaudi-docker/1.17.0/ubuntu22.04/habanalabs/pytorch-installer-2.3.1:latest` base image, make sure to replace it with the appropriate version for your environment if needed.
+See the [PyTorch Docker Images for the Intel® Gaudi® Accelerator](https://developer.habana.ai/catalog/pytorch-container/) for more information.
+
 ### Run the Container
 
 ```bash