diff --git a/Dockerfile b/Dockerfile
index 83b9215..ff8656e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.2-base-ubuntu18.04
+FROM nvidia/cuda:11.2.1-base-ubuntu18.04
 
 LABEL maintainer="dockerfile@codingcoffee.me"