diff options
author | Mathieu Poirier <mathieu.poirier@linaro.org> | 2021-02-05 11:46:15 -0700 |
---|---|---|
committer | Mathieu Poirier <mathieu.poirier@linaro.org> | 2021-02-05 11:46:15 -0700 |
commit | 722d2adad8a50846e1e0151f2c52795b778821a0 (patch) | |
tree | f8dbaa97dfb9b6243cfb187368f5d807c8999cc0 | |
parent | 57f190c14d32cf7f6a0d8453575aa097036a44cb (diff) |
Initial image to support deeplearning
Signed-off-by: Mathieu Poirier <mathieu.poirier@linaro.org>
-rw-r--r-- | patchnet-deeplearning | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/patchnet-deeplearning b/patchnet-deeplearning new file mode 100644 index 0000000..cc182dc --- /dev/null +++ b/patchnet-deeplearning @@ -0,0 +1,40 @@ +FROM tensorflow/tensorflow:latest-gpu + +RUN apt-get update + +RUN mkdir -p /home/linaro +WORKDIR /home/linaro + +# Install git +RUN apt-get install -y git + +# Install python 3.8 +RUN add-apt-repository -y ppa:deadsnakes/ppa +RUN apt-get install -y python3.8 +RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 0 +RUN update-alternatives --set python3 /usr/bin/python3.8 + +# Upgrade pip +RUN python -m pip install --user --upgrade pip + +# Install python compatibility library +RUN pip install six + +# +# Deeplearning dependencies +# + +# Install tqdm +RUN pip install tqdm + +# Install skikit-learn +RUN pip install scikit-learn + +# Install torch +# +# Use the grid on this site to costumise this line: https://pytorch.org/get-started/locally/ +RUN pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html + +# Install the PatchNet model +RUN git clone https://github.com/Xin-Zhou-smu/PatchNet.git + |