Skip to content

Commit 82fcfce

Browse files
ErnevSharmashaernev
andauthored
Huggingface pytorch training latest (#4687)
* chore: manually installing latest Go version * Update ['dlc_developer_config.toml'] dlc_developer_config.toml: { 'build': { 'build_frameworks': ['huggingface_pytorch'], 'build_inference': False, 'build_training': True}, 'buildspec_override': { 'dlc-pr-huggingface-pytorch-training': 'huggingface/pytorch/training/buildspec.yml'}, 'dev': { 'arm64_mode': False, 'deep_canary_mode': False, 'graviton_mode': False, 'neuronx_mode': False}, 'test': { 'ec2_tests': True, 'ecs_tests': True, 'eks_tests': True, 'sagemaker_local_tests': True, 'sagemaker_remote_tests': True, 'sanity_tests': True, 'security_tests': True}} * Revert "Update ['dlc_developer_config.toml']" This reverts commit edc424d. --------- Co-authored-by: shaernev <shaernev@amazon.com>
1 parent 3e45211 commit 82fcfce

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

huggingface/pytorch/training/docker/2.5/py3/cu124/Dockerfile.gpu

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,10 +56,16 @@ ENV HF_HUB_ENABLE_HF_TRANSFER="1"
5656
RUN apt-get update \
5757
# TODO: Remove upgrade statements once packages are updated in base image
5858
&& apt-get -y upgrade --only-upgrade systemd openssl cryptsetup libkrb5-3 \
59-
&& apt install -y git git-lfs golang-go \
59+
&& apt-get install -y git git-lfs wget tar \
60+
&& wget https://go.dev/dl/go1.22.2.linux-amd64.tar.gz \
61+
&& rm -rf /usr/local/go \
62+
&& tar -C /usr/local -xzf go1.22.2.linux-amd64.tar.gz \
63+
&& ln -s /usr/local/go/bin/go /usr/bin/go \
64+
&& rm go1.22.2.linux-amd64.tar.gz \
6065
&& apt-get clean \
6166
&& rm -rf /var/lib/apt/lists/*
6267

68+
6369
COPY cuda-compatibility-lib.sh /usr/local/bin/cuda-compatibility-lib.sh
6470
RUN chmod +x /usr/local/bin/cuda-compatibility-lib.sh
6571

0 commit comments

Comments
 (0)