fix docker tag for cuda (#2952)

deepset-ai · Aug 3, 2022 · 3728a95 · 3728a95
1 parent 7b97bbb
commit 3728a95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile-GPU b/Dockerfile-GPU
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.1-runtime-ubuntu20.04
+FROM nvidia/cuda:11.1.1-runtime-ubuntu20.04
 
 WORKDIR /home/user