Skip to content

Commit

Permalink
remove explicit image pull for aks-gpu
Browse files Browse the repository at this point in the history
  • Loading branch information
ganeshkumarashok committed Oct 23, 2024
1 parent 054f90b commit d24adfe
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions vhdbuilder/packer/install-dependencies.sh
Original file line number Diff line number Diff line change
Expand Up @@ -353,12 +353,9 @@ capture_benchmark "${SCRIPT_NAME}_artifact_streaming_download"

if [[ $OS == $UBUNTU_OS_NAME && $(isARM64) != 1 ]]; then # no ARM64 SKU with GPU now
gpu_action="copy"
NVIDIA_DRIVER_IMAGE_SHA="20241008175307"
export NVIDIA_DRIVER_IMAGE_TAG="550.90.12-${NVIDIA_DRIVER_IMAGE_SHA}"
NVIDIA_DRIVER_IMAGE="mcr.microsoft.com/aks/aks-gpu-cuda"


mkdir -p /opt/{actions,gpu}
ctr -n k8s.io image pull $NVIDIA_DRIVER_IMAGE:$NVIDIA_DRIVER_IMAGE_TAG
if grep -q "fullgpu" <<< "$FEATURE_FLAGS"; then
bash -c "$CTR_GPU_INSTALL_CMD $NVIDIA_DRIVER_IMAGE:$NVIDIA_DRIVER_IMAGE_TAG gpuinstall /entrypoint.sh install"
ret=$?
Expand Down

0 comments on commit d24adfe

Please sign in to comment.