forked from dwiepert/naip-w2v2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
32 lines (23 loc) · 756 Bytes
/
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
29
30
31
32
# use a Google maintained base image hosted in
# Google's container registry
FROM gcr.io/deeplearning-platform-release/pytorch-gpu.1-13.py310
# package dependencies - install requirements not included in base image
ARG AIF_PIP_INDEX
RUN pip install -i $AIF_PIP_INDEX --upgrade pip
COPY requirements.txt requirements.txt
RUN pip install -r requirements.txt
#if you want to install with conda
#RUN conda install -y package_name=version
# copy all necessary code
#1. make src directory
RUN mkdir /src
#2. set src as working dir
WORKDIR /src
#3. copy all files from src
COPY ./src /src
#4. set python path
RUN export PYTHONPATH=/src/
#RUN dir # for checking file list
# execute the code
#CMD ["python", "run.py"]
ENTRYPOINT ["python", "run.py"]