From f71dd1775731a8049060225f183027ab02a8d0b9 Mon Sep 17 00:00:00 2001
From: kthui <18255193+kthui@users.noreply.github.com>
Date: Fri, 8 Dec 2023 13:37:05 -0800
Subject: [PATCH] Do not specify pytorch cuda version

---
 tools/gen_pb_exec_env.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/gen_pb_exec_env.sh b/tools/gen_pb_exec_env.sh
index 26914b6..17d22b0 100755
--- a/tools/gen_pb_exec_env.sh
+++ b/tools/gen_pb_exec_env.sh
@@ -41,7 +41,7 @@ export PYTHONNOUSERSITE=True
 conda install -c conda-forge libstdcxx-ng=12 -y
 
 # install PyTorch
-conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y
+conda install pytorch torchvision torchaudio pytorch-cuda -c pytorch -c nvidia -y
 
 # pack environment
 rm -f pb_exec_env_model.py.tar.gz