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