Update CUDA version

FloopCZ · FloopCZ · commit 11a1b596e8d4 · 2020-10-16T17:44:44.000+02:00
diff --git a/Dockerfiles/archlinux-cuda b/Dockerfiles/archlinux-cuda
@@ -10,4 +10,4 @@ RUN ./tensorflow_cc/Dockerfiles/install-archlinux.sh --cuda
 
 ENV NVIDIA_VISIBLE_DEVICES all
 ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
-ENV NVIDIA_REQUIRE_CUDA "cuda>=10.1"
+ENV NVIDIA_REQUIRE_CUDA "cuda>=11.1"
diff --git a/Dockerfiles/ubuntu-cuda b/Dockerfiles/ubuntu-cuda
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
+FROM nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04
 
 # copy the contents of this repository to the container
 COPY . tensorflow_cc

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04`
	`1`	`+FROM nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04`
`2`	`2`
`3`	`3`	`# copy the contents of this repository to the container`
`4`	`4`	`COPY . tensorflow_cc`