Skip to content

Commit e9ff7c9

Browse files
committed
fix: L0_pytorch_python_runtime
Attempt to fix test failing from dependency incompatabilities.
1 parent 31a1202 commit e9ff7c9

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tools/gen_pb_exec_env.sh

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,13 +37,15 @@ conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
3737
conda create -n pt python=3.12 -y
3838
conda activate pt
3939
conda install -c conda-forge conda-pack -y
40+
conda install pip
4041

4142
# pre install step
4243
export PYTHONNOUSERSITE=True
4344
conda install -c conda-forge libstdcxx-ng=15 -y
4445

45-
# install PyTorch
46-
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia -y
46+
# install PyTorch (torch from pip to avoid conda version issues)
47+
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130
48+
conda install torchaudio pytorch-cuda=13.0 -c pytorch -c nvidia -y
4749

4850
# pack environment
4951
rm -f pb_exec_env_model.py.tar.gz

0 commit comments

Comments
 (0)