From 43e1004b15d115b69a789719921a2a18e52df565 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Thu, 18 Jul 2024 00:34:27 +0000
Subject: [PATCH] update

---
 spawnerdockerfile/Dockerfile.ngc | 1 +
 spawnerdockerfile/README.md      | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/spawnerdockerfile/Dockerfile.ngc b/spawnerdockerfile/Dockerfile.ngc
index 8399dc2..cd9c6db 100644
--- a/spawnerdockerfile/Dockerfile.ngc
+++ b/spawnerdockerfile/Dockerfile.ngc
@@ -35,6 +35,7 @@ apt-get install -y libnvidia-compute-535
 apt-get install -y flex tk ethtool libpci3 libltdl-dev bison lsof tcl libelf1 pciutils kmod libmnl0 debhelper libusb-1.0-0 graphviz chrpath swig libfuse2 udev
 # install Mellanox OFED
 mkdir -p ${STAGE_DIR}
+cd ${STAGE_DIR}
 wget -q -O - http://www.mellanox.com/downloads/ofed/MLNX_OFED-${MLNX_OFED_VERSION}/MLNX_OFED_LINUX-${MLNX_OFED_VERSION}-ubuntu22.04-x86_64.tgz | tar xzf -
 cd ${STAGE_DIR}/MLNX_OFED_LINUX-${MLNX_OFED_VERSION}-ubuntu22.04-x86_64
 ./mlnxofedinstall --user-space-only --without-fw-update --all -q > /tmp/mlnxofedinstall.log 2>&1
diff --git a/spawnerdockerfile/README.md b/spawnerdockerfile/README.md
index 2d9a111..c68e74a 100755
--- a/spawnerdockerfile/README.md
+++ b/spawnerdockerfile/README.md
@@ -30,7 +30,7 @@ NCCL_IB_DISABLE=0
 
 docker run -it --rm --network=host --privileged --ipc=host --ulimit memlock=-1 --gpus all hotwa/notebook:ngc
 docker run --rm -it --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 nvcr.io/nvidia/pytorch:24.06-py3 /bin/bash
-docker run --rm -it --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 hotwa/notebook:ngc /bin/bash
+docker run -it --rm --network=host --privileged --ipc=host --ulimit memlock=-1 --gpus all quay.io/hotwa/ngc:latest /bin/bash
 
 ```shell
 nvidia-smi