From b5f5462cd4af142b6f0bdb4b0fedefa82e450728 Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Tue, 15 Jul 2025 07:22:15 -0700
Subject: [PATCH] Test with CUDA 12.0

---
 .circleci/config.yml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index ef82298bc..9143b5467 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -208,7 +208,13 @@ jobs:
       - run:
           name: Install Python package
           command: |
+            wget http://security.ubuntu.com/ubuntu/pool/universe/n/ncurses/libtinfo5_6.3-2ubuntu0.1_amd64.deb
+            sudo dpkg -i libtinfo5_6.3-2ubuntu0.1_amd64.deb
+            wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb
+            sudo dpkg -i cuda-keyring_1.0-1_all.deb
             sudo apt-get update
+            sudo apt-get install cuda-toolkit-12-0
+            sudo update-alternatives --set cuda /usr/local/cuda-12.0
             sudo apt-get install libblas-dev liblapack-dev liblapacke-dev
             python -m venv env
             source env/bin/activate