mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-03-14 20:27:24 +00:00
48 lines
1.6 KiB
Bash
Executable File
48 lines
1.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
setup_python_env() {
|
|
local py_version=$1
|
|
|
|
# check if pyenv is installed
|
|
if ! command -v pyenv &> /dev/null; then
|
|
rm -f /pyenv
|
|
curl -fsSL https://pyenv.run | bash
|
|
fi
|
|
|
|
# Install the build dependencies, check /etc/os-release to see if we are on ubuntu or rocky
|
|
if [ -f /etc/os-release ]; then
|
|
source /etc/os-release
|
|
if [ "$ID" = "ubuntu" ]; then
|
|
# Use the retry helper to mitigate issues with apt network errors:
|
|
script_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
|
|
retry() {
|
|
"${script_dir}/util/retry.sh" 5 30 "$@"
|
|
}
|
|
|
|
retry sudo apt update
|
|
retry sudo apt install -y make libssl-dev zlib1g-dev \
|
|
libbz2-dev libreadline-dev libsqlite3-dev curl git \
|
|
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev
|
|
elif [ "$ID" = "rocky" ]; then
|
|
# we're inside the rockylinux container, sudo not required/available
|
|
dnf install -y make patch zlib-devel bzip2 bzip2-devel readline-devel \
|
|
sqlite sqlite-devel openssl-devel tk-devel libffi-devel xz-devel libuuid-devel \
|
|
gdbm-libs libnsl2
|
|
else
|
|
echo "Unsupported Linux distribution"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
# Always set up pyenv environment
|
|
export PYENV_ROOT="$HOME/.pyenv"
|
|
[[ -d $PYENV_ROOT/bin ]] && export PATH="$PYENV_ROOT/bin:$PATH"
|
|
eval "$(pyenv init - bash)"
|
|
|
|
# Using pyenv, install the Python version
|
|
PYENV_DEBUG=1 pyenv install -v "${py_version}"
|
|
pyenv local "${py_version}"
|
|
|
|
pip install --upgrade pip
|
|
}
|