Skip to content

Commit

Permalink
Merge pull request Azure#738 from Azure/enroot-pyxis_no_download
Browse files Browse the repository at this point in the history
Install pyxis and enroot from local packages
  • Loading branch information
yosoyjay authored Dec 21, 2023
2 parents b8aef25 + 7a7e85b commit a104571
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 19 deletions.
Binary file not shown.
Binary file not shown.
Original file line number Diff line number Diff line change
Expand Up @@ -2,24 +2,16 @@

source $CYCLECLOUD_SPEC_PATH/files/common_functions.sh

ENROOT_VERSION_FULL=${1:-3.4.1-1}
ENROOT_VERSION=${ENROOT_VERSION_FULL%-*}
ENROOT_VERSION='3.4.1-1'

# Install enroot RPM packages on compute nodes
if ! is_slurm_controller; then

cd /tmp

# Install NVIDIA enroot
curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v${ENROOT_VERSION}/enroot_${ENROOT_VERSION_FULL}_amd64.deb
curl -fSsL -O https://github.com/NVIDIA/enroot/releases/download/v${ENROOT_VERSION}/enroot+caps_${ENROOT_VERSION_FULL}_amd64.deb
apt install -y ./enroot_${ENROOT_VERSION_FULL}_amd64.deb
apt install -y ./enroot+caps_${ENROOT_VERSION_FULL}_amd64.deb
apt install -y $CYCLECLOUD_SPEC_PATH/files/enroot_${ENROOT_VERSION}_amd64.deb
apt install -y $CYCLECLOUD_SPEC_PATH/files/enroot+caps_${ENROOT_VERSION}_amd64.deb

# Install NVIDIA container support
apt-get install -y libnvidia-container1 libnvidia-container-tools

rm -f /tmp/enroot_${ENROOT_VERSION_FULL}_amd64.deb
rm -f /tmp/enroot+caps_${ENROOT_VERSION_FULL}_amd64.deb

fi
Original file line number Diff line number Diff line change
Expand Up @@ -10,15 +10,10 @@ function install_build_deps() {
apt-get -y install gcc make
}

function get_source() {
wget https://github.com/NVIDIA/pyxis/archive/refs/tags/v${PYXIS_VER}.tar.gz
tar xzf v${PYXIS_VER}.tar.gz
}

function install_from_source() {
install_build_deps
cd ${TMP_DIR}
get_source
tar -xzf $CYCLECLOUD_SPEC_PATH/files/pyxis_${PYXIS_VER}.tar.gz
cd pyxis-${PYXIS_VER}
make install

Expand All @@ -34,8 +29,7 @@ function install_from_source() {
chmod +x /usr/lib64/slurm/spank_pyxis.so

cd ~
rm -rf ${TMP_DIR}/v${PYXIS_VER}.tar.gz \
${TMP_DIR}/pyxis-${PYXIS_VER}
rm ${TMP_DIR}/pyxis-${PYXIS_VER}
}

function install_compute_node() {
Expand Down
Binary file not shown.

0 comments on commit a104571

Please sign in to comment.