From 53816a0bfddd403822fc41a2f1e9f631a2cee3af Mon Sep 17 00:00:00 2001 From: EveningStarlight Date: Wed, 8 Jan 2025 09:15:47 -0500 Subject: [PATCH] add pytorch requirments file --- images/base/Dockerfile | 3 +++ images/base/pytorch-requirements.txt | 10 ++++++++++ 2 files changed, 13 insertions(+) create mode 100644 images/base/pytorch-requirements.txt diff --git a/images/base/Dockerfile b/images/base/Dockerfile index c5530f687..d3ba07a5c 100644 --- a/images/base/Dockerfile +++ b/images/base/Dockerfile @@ -12,6 +12,9 @@ RUN apt-get update --yes \ COPY aaw-suspend-server.sh /usr/local/bin +# Add requirments file for adding pytorch +COPY pytorch-requirements.txt /usr/local/bin/requirments.txt + # https://github.com/StatCan/aaw-kubeflow-containers/issues/293 RUN mamba install --quiet \ 'pillow' \ diff --git a/images/base/pytorch-requirements.txt b/images/base/pytorch-requirements.txt new file mode 100644 index 000000000..393b49d3d --- /dev/null +++ b/images/base/pytorch-requirements.txt @@ -0,0 +1,10 @@ +# Requirments file for adding pytorch to image + +ipykernel +pytorch +torchvision +torchaudio +# gputil has nvidia-smi +gputil +# pytorch-cuda are the nvidia cuda drivers +pytorch-cuda == 11.8 \ No newline at end of file