Adding back gpu for testing on magnet 5

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