From c2ea6b748af780116a31f94f376e1d9a3b100815 Mon Sep 17 00:00:00 2001 From: Keita Watanabe Date: Mon, 25 Mar 2024 17:45:21 +0900 Subject: [PATCH] Update 3.container-train.sbatch --- 3.test_cases/16.pytorch-cpu-ddp/3.container-train.sbatch | 1 + 1 file changed, 1 insertion(+) diff --git a/3.test_cases/16.pytorch-cpu-ddp/3.container-train.sbatch b/3.test_cases/16.pytorch-cpu-ddp/3.container-train.sbatch index 4109a3c6..636ddfff 100644 --- a/3.test_cases/16.pytorch-cpu-ddp/3.container-train.sbatch +++ b/3.test_cases/16.pytorch-cpu-ddp/3.container-train.sbatch @@ -17,6 +17,7 @@ head_node_ip=$(srun --nodes=1 --ntasks=1 -w "$head_node" hostname --ip-address) echo Node IP: $head_node_ip export LOGLEVEL=INFO +export NVIDIA_VISIBLE_DEVICES=void declare -a ARGS=( --container-image ${PWD}/pytorch.sqsh