linux-cuda image

3a0c9c8582e23f14dcd887d28148428b834f42e6 · 2024-08-02 14:49:07 +0100 · Alex Cheema

Files touched

M .circleci/config.yml

Diff

commit 3a0c9c8582e23f14dcd887d28148428b834f42e6
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Fri Aug 2 14:49:07 2024 +0100

    linux-cuda image
---
 .circleci/config.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index 6759ab13..4f0cd9b2 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -173,10 +173,11 @@ jobs:
 
   chatgpt_api_integration_test_tinygrad:
     machine:
-      image: ubuntu-2204-cuda-11.8:current
+      image: linux-cuda-12:default
     resource_class: gpu.nvidia.small.gen2
     steps:
       - checkout
+      - run: nvidia-smi
       - run:
           name: Set up Python
           command: |

← af2f98ba run tinygrad tests on gpu.nvidia.small.gen2 (NVIDIA A10G 24G · back to Exo · use gpu.nvidia.medium a3b0650f →