Added GPU support to docker container.

Makefile

@@ -1,12 +1,12 @@
 all: benchmark statistics
 
 benchmark: docker/container.ok
-	docker container run --rm -it -v `(pwd)`:/benchmark/data convganbenchmark python3 /benchmark/data/run_all_exercises.py
-	docker container run --rm -it -v `(pwd)`:/benchmark/data convganbenchmark chown -R `(./getMyUid) /benchmark/data/data_results
+	docker container run --gpus all --rm -it -v `(pwd)`:/benchmark/data convganbenchmark python3 /benchmark/data/run_all_exercises.py
+	docker container run --rm -it -v `(pwd)`:/benchmark/data convganbenchmark chown -R `(./getMyUid)` /benchmark/data/data_results
 
 
 statistics: docker/container.ok
-	docker container run --rm -it -v `(pwd)`:/benchmark/data convganbenchmark python3 /benchmark/data/statistics.py
+	docker container run --gpus all --rm -it -v `(pwd)`:/benchmark/data convganbenchmark python3 /benchmark/data/statistics.py
 	docker container run --rm -it -v `(pwd)`:/benchmark/data convganbenchmark chown -R `(./getMyUid)` /benchmark/data/data_result
 
 docker/container.ok: docker/Dockerfile docker/run.sh docker/requirements.txt

docker/Dockerfile

@@ -1,4 +1,6 @@
-FROM ubuntu:21.10
+
+#FROM ubuntu:21.10
+FROM nvidia/cuda:11.3.1-cudnn8-devel
 
 RUN ln -s /usr/share/zoneinfo/Europe/Berlin /etc/localtime
 RUN apt-get update && apt-get dist-upgrade -y && apt-get install -y python3 pip && apt-get clean

docker/requirements.txt

@@ -1,11 +1,11 @@
-tensorflow==2.7.0
-torch==1.10.2
-scikit-image==0.18.3
-scikit-learn==0.24.2
-tqdm==4.62.3
-matplotlib==3.4.3
+tensorflow>=2.7.0
+torch>=1.10.2
+scikit-image>=0.18.3
+scikit-learn>=0.24.2
+tqdm>=4.62.3
+matplotlib>=3.4.3
 imblearn
-numpy==1.20.3
-ctgan==0.5.1
-keras==2.7.0
+numpy>=1.20.3
+ctgan>=0.5.1
+keras>=2.7.0
 pickleshare>=0.7.5