diff --git a/tools/gen_pb_exec_env.sh b/tools/gen_pb_exec_env.sh index d7fe86b..d01a0d9 100755 --- a/tools/gen_pb_exec_env.sh +++ b/tools/gen_pb_exec_env.sh @@ -27,18 +27,18 @@ # install conda rm -rf ./miniconda -wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.3.1-0-Linux-x86_64.sh -bash Miniconda3-py310_23.3.1-0-Linux-x86_64.sh -p ./miniconda -b +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 eval "$(./miniconda/bin/conda shell.bash hook)" # create conda environment -conda create -n pt python=3.10 -y +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=12 -y +conda install -c conda-forge libstdcxx-ng=14 -y # install PyTorch conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y @@ -49,4 +49,3 @@ conda pack -o pb_exec_env_model.py.tar.gz # deactivate conda conda deactivate -conda deactivate