Skip to content
This repository has been archived by the owner on Mar 20, 2023. It is now read-only.

Commit

Permalink
Fix nvidia-docker2 installation
Browse files Browse the repository at this point in the history
- Properly pin dependencies
- Resolves #231
- Update bug report template
  • Loading branch information
alfpark committed Jul 20, 2018
1 parent aa3b7f1 commit 85db792
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 2 deletions.
2 changes: 2 additions & 0 deletions .github/ISSUE_TEMPLATE/bug_report.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@ about: Create a report to help us improve

#### Problem Description

#### Batch Shipyard Version

#### Steps to Reproduce

#### Expected Results
Expand Down
7 changes: 5 additions & 2 deletions scripts/shipyard_nodeprep.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ set -o pipefail
DOCKER_CE_VERSION_DEBIAN=18.03.1
DOCKER_CE_VERSION_CENTOS=18.03.1
DOCKER_CE_VERSION_SLES=17.09.1
NVIDIA_CONTAINER_RUNTIME_VERSION=2.0.0
NVIDIA_DOCKER_VERSION=2.0.3
GLUSTER_VERSION_DEBIAN=4.1
GLUSTER_VERSION_CENTOS=41
Expand All @@ -18,6 +19,8 @@ DOCKER_CE_PACKAGE_DEBIAN_OLD="docker-ce=${DOCKER_CE_VERSION_DEBIAN}~ce-0~"
DOCKER_CE_PACKAGE_DEBIAN_NEW="docker-ce=${DOCKER_CE_VERSION_DEBIAN}~ce~3-0~"
DOCKER_CE_PACKAGE_CENTOS="docker-ce-${DOCKER_CE_VERSION_CENTOS}.ce-1.el7.centos"
DOCKER_CE_PACKAGE_SLES="docker-${DOCKER_CE_VERSION_SLES}_ce-257.3"
NVIDIA_CONTAINER_RUNTIME_PACKAGE_UBUNTU="nvidia-container-runtime=${NVIDIA_CONTAINER_RUNTIME_VERSION}+docker${DOCKER_CE_VERSION_DEBIAN}-1"
NVIDIA_CONTAINER_RUNTIME_PACKAGE_CENTOS="nvidia-container-runtime-${NVIDIA_CONTAINER_RUNTIME_VERSION}-1.docker${DOCKER_CE_VERSION_CENTOS}.ce"
NVIDIA_DOCKER_PACKAGE_UBUNTU="nvidia-docker2=${NVIDIA_DOCKER_VERSION}+docker${DOCKER_CE_VERSION_DEBIAN}-1"
NVIDIA_DOCKER_PACKAGE_CENTOS="nvidia-docker2-${NVIDIA_DOCKER_VERSION}-1.docker${DOCKER_CE_VERSION_CENTOS}.ce"
MOUNTS_PATH=$AZ_BATCH_NODE_ROOT_DIR/mounts
Expand Down Expand Up @@ -604,9 +607,9 @@ EOF
fi
refresh_package_index
if [ "$DISTRIB_ID" == "ubuntu" ]; then
install_packages "$NVIDIA_DOCKER_PACKAGE_UBUNTU"
install_packages "$NVIDIA_CONTAINER_RUNTIME_PACKAGE_UBUNTU" "$NVIDIA_DOCKER_PACKAGE_UBUNTU"
elif [[ $DISTRIB_ID == centos* ]]; then
install_packages "$NVIDIA_DOCKER_PACKAGE_CENTOS"
install_packages "$NVIDIA_CONTAINER_RUNTIME_PACKAGE_CENTOS" "$NVIDIA_DOCKER_PACKAGE_CENTOS"
fi
# merge daemon configs if necessary
set +e
Expand Down

0 comments on commit 85db792

Please sign in to comment.