test cuda

23d2432cb91b183dce15045d32fc8b10f9544efa · 2024-08-02 16:03:58 +0100 · Alex Cheema

Files touched

M .circleci/config.yml

Diff

commit 23d2432cb91b183dce15045d32fc8b10f9544efa
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Fri Aug 2 16:03:58 2024 +0100

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

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 22869d42..3db98e20 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -172,12 +172,12 @@ jobs:
           model_id: llama-3.1-8b
 
   test_cuda:
-    docker:
-      - image: cimg/base:current
-    resource_class: xlarge
+    machine:
+      image: linux-cuda-12:default
+    resource_class: gpu.nvidia.medium
     steps:
       - checkout
-      - run: echo "hi"
+      - run: nvidia-smi
 
   chatgpt_api_integration_test_tinygrad:
     docker:
@@ -186,7 +186,7 @@ jobs:
     steps:
       - checkout
       - run:
-          name: Set up Python
+          name: Set up
           command: |
             sudo add-apt-repository ppa:deadsnakes/ppa -y
             sudo apt update -y

← fcdf57b8 t · back to Exo · cuda test e10aa581 →