diff --git a/.circleci/config.yml b/.circleci/config.yml
index abcb1ad..4af111e 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -4,7 +4,7 @@ version: 2.1
 jobs:
   build:
     machine:
-      image: linux-cuda-11:default
+      image: gpu.nvidia.small.multi
 #    docker:
 #      - image: cimg/python:3.8  # You can change the Python version as needed
     steps: