cuda test

e10aa581ac4b85551fa1632d4798529a55b9b81c · 2024-08-02 16:06:54 +0100 · Alex Cheema

Files touched

M .circleci/config.yml

Diff

commit e10aa581ac4b85551fa1632d4798529a55b9b81c
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Fri Aug 2 16:06:54 2024 +0100

     cuda test
---
 .circleci/config.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 3db98e20..f57cab64 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -172,8 +172,8 @@ jobs:
           model_id: llama-3.1-8b
 
   test_cuda:
-    machine:
-      image: linux-cuda-12:default
+    docker:
+      - image: nvidia/cuda:11.4.3-base-ubuntu20.04
     resource_class: gpu.nvidia.medium
     steps:
       - checkout

← 23d2432c test cuda · back to Exo · mac runners 0eb5c0c6 →