diff --git a/examples/heart-uci/docker.sh b/examples/heart-uci/docker.sh
index 8cee5122e74b1429c569cade97feb990d62d4f63..5a2f8185f14be573f0c9a76a226f6bfb6b4741de 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