diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index ea19b6ebd3652aef6dd5218f1896f545616394d9..4067b3edc562a892e3af04612b8d5b4bd1904398 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,5 +1,5 @@
 build:
-  image: elixxirlabs/cuda-go:go1.16-cuda11.1
+  image: $DOCKER_IMAGE
   script:
     - GOOS=linux GOARCH=amd64 CGO_ENABLED=0 go build -ldflags '-w -s' -o release/server.linux64 main.go
     - GOOS=windows GOARCH=amd64 CGO_ENABLED=0 go build -ldflags '-w -s' -o release/server.win64 main.go