From 30fcd4f9ff0c53306bc399605c131c8b6b6701e5 Mon Sep 17 00:00:00 2001
From: Blake Mallory <blakecmallory@gmail.com>
Date: Wed, 27 Jul 2022 00:24:52 -0400
Subject: [PATCH] updated docker file and added script to instsall needed
 package post image

---
 Dockerfile                       | 11 ++++++++++-
 install_image_local_attention.sh |  3 +++
 2 files changed, 13 insertions(+), 1 deletion(-)
 create mode 100755 install_image_local_attention.sh

diff --git a/Dockerfile b/Dockerfile
index 1b95eca..239f832 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,6 +3,9 @@ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
 ARG DEBIAN_FRONTEND=noninteractive
 ENV TZ=Europe/London
 
+SHELL ["/bin/bash", "-c"]
+
+
 RUN apt-get update && apt-get upgrade -y
 RUN apt install git -y \
 && apt install python3 -y \
@@ -10,8 +13,14 @@ RUN apt install git -y \
 && apt install git -y \
 && apt install python-is-python3 -y \
 && apt install python3-tk -y \
-&& apt install ffmpeg libsm6 libxext6  -y
+&& apt install ffmpeg libsm6 libxext6  -y \ 
+&& apt install git -y
+
+
+WORKDIR /workspace
 
 RUN pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 \
 && pip install opencv-python
 
+
+RUN pip install SwissArmyTransformer>=0.2.9 icetk gifmaker
diff --git a/install_image_local_attention.sh b/install_image_local_attention.sh
new file mode 100755
index 0000000..df26da0
--- /dev/null
+++ b/install_image_local_attention.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+
+pip install git+https://github.com/Sleepychord/Image-Local-Attention
\ No newline at end of file