diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 53178d2..523bc3c 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.2.0-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
+FROM nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
 
 RUN apt-get update
 RUN apt-get install -y build-essential curl python3 python3-pip python3-venv