From 5bc662d35268989cb6fe6425f4e4e0bb9f223ec4 Mon Sep 17 00:00:00 2001 From: Yingge He <157551214+yinggeh@users.noreply.github.com> Date: Thu, 25 Sep 2025 09:53:30 -0700 Subject: [PATCH] fix: L0_pytorch_python_runtime--PyTorch (#156) --- tools/gen_pb_exec_env.sh | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tools/gen_pb_exec_env.sh b/tools/gen_pb_exec_env.sh index d01a0d9..19539cd 100755 --- a/tools/gen_pb_exec_env.sh +++ b/tools/gen_pb_exec_env.sh @@ -1,5 +1,5 @@ #!/bin/bash -# Copyright 2023-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright 2023-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions @@ -27,21 +27,23 @@ # install conda rm -rf ./miniconda -wget https://repo.anaconda.com/miniconda/Miniconda3-py312_24.9.2-0-Linux-x86_64.sh -bash Miniconda3-py312_24.9.2-0-Linux-x86_64.sh -p ./miniconda -b +wget https://repo.anaconda.com/miniconda/Miniconda3-py312_25.7.0-2-Linux-x86_64.sh +bash Miniconda3-py312_25.7.0-2-Linux-x86_64.sh -p ./miniconda -b eval "$(./miniconda/bin/conda shell.bash hook)" # create conda environment +conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main +conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r conda create -n pt python=3.12 -y conda activate pt conda install -c conda-forge conda-pack -y # pre install step export PYTHONNOUSERSITE=True -conda install -c conda-forge libstdcxx-ng=14 -y +conda install -c conda-forge libstdcxx-ng=15 -y # install PyTorch -conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y +conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia -y # pack environment rm -f pb_exec_env_model.py.tar.gz