-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathDockerfile
28 lines (23 loc) · 1.02 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# syntax=docker/dockerfile:1
# python version 3.6
# FROM waggle/plugin-base:1.1.1-ml-torch1.9
# needs this version for the pytorch 1.13.1 with cuda 11.6
FROM python:3.10.14-bullseye
# CUDA supports backward compatibility, so the application version <= driver version
# 11.8 for Dell blade server, 10.2 for waggle node
# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-24-03.html
RUN apt-get update && apt-get install -y \
ffmpeg libsm6 libxext6 libhdf5-dev redis-tools
WORKDIR /app
COPY requirements.txt requirements.txt
RUN pip install -r requirements.txt
# see this for torch installation https://pytorch.org/get-started/previous-versions/#v1131
RUN pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --index-url https://download.pytorch.org/whl/cu116
RUN pip install pywaggle[all]
RUN pip install h5py
# fix numpy compatibility issue
RUN pip install "numpy<2"
# native pyymal version is too old with the python
# RUN pip install --ignore-installed PyYAML
# COPY . .
CMD ["/bin/sh", "-c", "bash"]