From a207a34bd7632eddab5a97f6b8850bcb2bc42fe9 Mon Sep 17 00:00:00 2001
From: Ruida Zeng <31152346+ruidazeng@users.noreply.github.com>
Date: Thu, 13 Feb 2025 09:51:50 -0600
Subject: [PATCH] docs: updated to CUDA 12.4 (#51)

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index bced667..b33e967 100644
--- a/README.md
+++ b/README.md
@@ -23,8 +23,8 @@ The dataset is in QA format, making it ideal for use with popular chat models su
 ```
 conda create -n tofu python=3.10
 conda activate tofu
-conda install pytorch pytorch-cuda=11.8 -c pytorch -c nvidia
-conda install -c "nvidia/label/cuda-11.8.0" cuda-toolkit
+conda install pytorch pytorch-cuda=12.4 -c pytorch -c nvidia
+conda install -c "nvidia/label/cuda-12.4.0" cuda-toolkit
 pip install -r requirements.txt
 pip install flash-attn --no-build-isolation
 ```