From 3f22b6e6025ef60fe04fd97c31ed302dbc5811dc Mon Sep 17 00:00:00 2001
From: BIGAUD Nathan <nathan.bigaud@inria.fr>
Date: Mon, 22 May 2023 15:37:59 +0200
Subject: [PATCH] Adding back gpu for testing on magnet 5
---
examples/heart-uci/docker.sh | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/heart-uci/docker.sh b/examples/heart-uci/docker.sh
index 8cee5122..5a2f8185 100644
--- a/examples/heart-uci/docker.sh
+++ b/examples/heart-uci/docker.sh
@@ -5,7 +5,7 @@ cat <<- EOF > docker_server.sh
python gen_ssl.py
python server.py 4
EOF
-docker run -dt --network="host" --mount type=bind,source=$PWD,target=/experiment --name server --env script=./docker_server.sh registry.gitlab.inria.fr/magnet/declearn/declearn2:declearn-base
+docker run -dt --network="host" --gpus all --mount type=bind,source=$PWD,target=/experiment --name server --env script=./docker_server.sh registry.gitlab.inria.fr/magnet/declearn/declearn2:declearn-base
# Ensure server is done generating certificates
sleep 2
# Run client scripts and run containers
@@ -14,7 +14,7 @@ do
cat <<- EOF > docker_$CLIENT.sh
python client.py $CLIENT
EOF
- docker run -dt --network=container:server --mount type=bind,source=$PWD,target=/experiment --name $CLIENT --env script=./docker_$CLIENT.sh registry.gitlab.inria.fr/magnet/declearn/declearn2:declearn-base
+ docker run -dt --network=container:server --gpus all --mount type=bind,source=$PWD,target=/experiment --name $CLIENT --env script=./docker_$CLIENT.sh registry.gitlab.inria.fr/magnet/declearn/declearn2:declearn-base
done
--
GitLab