From 25e4c4077e63787cc471530b002d03d434ed1573 Mon Sep 17 00:00:00 2001
From: Ean Garvey <87458719+monorimet@users.noreply.github.com>
Date: Tue, 28 May 2024 18:25:54 -0500
Subject: [PATCH] Update setup_venv.sh

---
 setup_venv.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup_venv.sh b/setup_venv.sh
index 9578cefb..f4aad89c 100755
--- a/setup_venv.sh
+++ b/setup_venv.sh
@@ -54,7 +54,7 @@ PYTORCH_URL="https://download.pytorch.org/whl/nightly/cpu/"
 
 # Upgrade pip and install requirements.
 $PYTHON -m pip install --upgrade pip || die "Could not upgrade pip"
-$PYTHON -m pip install --upgrade --pre torch torchvision torchaudio --index-url $PYTORCH_URL
+$PYTHON -m pip install --upgrade --pre torch==2.3.0 torchvision torchaudio --index-url $PYTORCH_URL
 if [[ -z "${NO_BREVITAS}" ]]; then
   $PYTHON -m pip install git+https://github.com/Xilinx/brevitas.git@dev
 fi