From b0e68540c47409e945455e0491c71be4673cc9fe Mon Sep 17 00:00:00 2001 From: Kalana Ratnayake Date: Thu, 8 Aug 2024 11:45:57 +1000 Subject: [PATCH] fixing an issue in pytorch docker images with cuda failure --- pytorch-images/r3630.Dockerfile | 31 ++++++++++++++++++++++++------- 1 file changed, 24 insertions(+), 7 deletions(-) diff --git a/pytorch-images/r3630.Dockerfile b/pytorch-images/r3630.Dockerfile index a70e71c..b4e4c57 100644 --- a/pytorch-images/r3630.Dockerfile +++ b/pytorch-images/r3630.Dockerfile @@ -13,23 +13,40 @@ WORKDIR / ###################################################################################### RUN apt-get update -y -RUN apt-get install -y --no-install-recommends python3-pip +RUN apt-get install -y --no-install-recommends python3-pip \ + libopenblas-dev + +RUN python3 -m pip install --no-cache-dir numpy ##################################################################################### -## Install PyTorch 2.4 +## Install PyTorch 2.3 ##################################################################################### -RUN wget https://developer.download.nvidia.cn/compute/redist/jp/v60/pytorch/torch-2.4.0a0+3bcc3cddb5.nv24.07.16234504-cp310-cp310-linux_aarch64.whl +RUN wget -O torch-2.3.0-cp310-cp310-linux_aarch64.whl https://nvidia.box.com/shared/static/mp164asf3sceb570wvjsrezk1p4ftj8t.whl -RUN python3 -m pip install --no-cache-dir torch-2.4.0a0+3bcc3cddb5.nv24.07.16234504-cp310-cp310-linux_aarch64.whl +RUN python3 -m pip install --no-cache-dir torch-2.3.0-cp310-cp310-linux_aarch64.whl -RUN rm torch-2.4.0a0+3bcc3cddb5.nv24.07.16234504-cp310-cp310-linux_aarch64.whl +RUN rm torch-2.3.0-cp310-cp310-linux_aarch64.whl ##################################################################################### -## Install PyTorch +## Install Torch Vision 0.18 ##################################################################################### -RUN python3 -m pip install --no-cache-dir torchvision torchaudio +RUN wget -O torchvision-0.18.0a0+6043bc2-cp310-cp310-linux_aarch64.whl https://nvidia.box.com/shared/static/xpr06qe6ql3l6rj22cu3c45tz1wzi36p.whl + +RUN python3 -m pip install --no-cache-dir torchvision-0.18.0a0+6043bc2-cp310-cp310-linux_aarch64.whl + +RUN rm torchvision-0.18.0a0+6043bc2-cp310-cp310-linux_aarch64.whl + +##################################################################################### +## Install Torch Audio 2.3 +##################################################################################### + +RUN wget -O torchaudio-2.3.0+952ea74-cp310-cp310-linux_aarch64.whl https://nvidia.box.com/shared/static/9agsjfee0my4sxckdpuk9x9gt8agvjje.whl + +RUN python3 -m pip install --no-cache-dir torchaudio-2.3.0+952ea74-cp310-cp310-linux_aarch64.whl + +RUN rm torchaudio-2.3.0+952ea74-cp310-cp310-linux_aarch64.whl ##################################################################################### ##