Update cuda version in wrapper Dockerfile.GPU

Bump up CUDA version from 10.0 to 10.2 and CudNN from 7 to 8 This should be enough to support `onnxruntime-gpu==1.6.0` with `torch==1.6.0` right out of the box
SeldonIO · Feb 1, 2021 · 03d6309 · 03d6309
1 parent f5f8731
commit 03d6309
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wrappers/s2i/python/Dockerfile.gpu b/wrappers/s2i/python/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
+FROM nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04
 
 LABEL io.openshift.s2i.scripts-url="image:///s2i/bin"