From a84eba57f92d1c179cb922aebc59a5d8577965fe Mon Sep 17 00:00:00 2001
From: Ben Walker <38265558+Benjamin-Walker@users.noreply.github.com>
Date: Mon, 4 Nov 2024 09:18:31 +0000
Subject: [PATCH] Update PyTorch environment to cuda 12.4

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

diff --git a/README.md b/README.md
index 21c6e7a..1ce821e 100644
--- a/README.md
+++ b/README.md
@@ -209,8 +209,8 @@ pre-commit install
 ```angular2html
 conda create -n pytorch_mamba python=3.11
 conda activate pytorch_mamba
-conda install pytorch=2.2 torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
-conda install packaging pyyaml -c conda-forge
+conda install pytorch=2.5 torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
+conda install packaging pyyaml pandas -c conda-forge
 pip install causal-conv1d>=1.2.0 mamba-ssm s5-pytorch einops
 ```