Skip to content

Commit

Permalink
test: Upgrade Miniconda version for ubuntu 24.04 (#142)
Browse files Browse the repository at this point in the history
* Update miniconda version

* For testing

* Try with python 3.12
  • Loading branch information
krishung5 authored and mc-nv committed Nov 23, 2024
1 parent 8fbacf5 commit b6f6d4a
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions tools/gen_pb_exec_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -49,4 +49,3 @@ conda pack -o pb_exec_env_model.py.tar.gz

# deactivate conda
conda deactivate
conda deactivate

0 comments on commit b6f6d4a

Please sign in to comment.