Skip to content

Commit db70751

Browse files
authored
fix: L0_pytorch_python_runtime--PyTorch (#156)
1 parent 8872dda commit db70751

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

tools/gen_pb_exec_env.sh

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2023-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2023-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -27,21 +27,23 @@
2727

2828
# install conda
2929
rm -rf ./miniconda
30-
wget https://repo.anaconda.com/miniconda/Miniconda3-py312_24.9.2-0-Linux-x86_64.sh
31-
bash Miniconda3-py312_24.9.2-0-Linux-x86_64.sh -p ./miniconda -b
30+
wget https://repo.anaconda.com/miniconda/Miniconda3-py312_25.7.0-2-Linux-x86_64.sh
31+
bash Miniconda3-py312_25.7.0-2-Linux-x86_64.sh -p ./miniconda -b
3232
eval "$(./miniconda/bin/conda shell.bash hook)"
3333

3434
# create conda environment
35+
conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main
36+
conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
3537
conda create -n pt python=3.12 -y
3638
conda activate pt
3739
conda install -c conda-forge conda-pack -y
3840

3941
# pre install step
4042
export PYTHONNOUSERSITE=True
41-
conda install -c conda-forge libstdcxx-ng=14 -y
43+
conda install -c conda-forge libstdcxx-ng=15 -y
4244

4345
# install PyTorch
44-
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y
46+
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia -y
4547

4648
# pack environment
4749
rm -f pb_exec_env_model.py.tar.gz

0 commit comments

Comments
 (0)