From 4ffe48ae0d35ac612326b45815bae071658e17cb Mon Sep 17 00:00:00 2001
From: Jeffrey Phillips Freeman <the@jeffreyfreeman.me>
Date: Thu, 1 Oct 2020 14:59:31 -0400
Subject: [PATCH] Cleaned out the tmp when docker image is done building and
 set it as the work dir.

---
 amdgpu/Dockerfile | 2 ++
 nvidia/Dockerfile | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/amdgpu/Dockerfile b/amdgpu/Dockerfile
index 57a9bfd..20a1fa0 100644
--- a/amdgpu/Dockerfile
+++ b/amdgpu/Dockerfile
@@ -38,4 +38,6 @@ RUN apt-get update && \
 ARG aparapiver
 RUN mvn org.apache.maven.plugins:maven-dependency-plugin:2.1:get -DrepoUrl="https://repo1.maven.org/maven2/" -Dartifact=com.aparapi:aparapi:${aparapiver}
 
+RUN rm -rf /tmp/*
+
 ENV JAVA_HOME /usr/lib/jvm/java-14-openjdk-amd64/
diff --git a/nvidia/Dockerfile b/nvidia/Dockerfile
index 124dd40..49e1ed2 100644
--- a/nvidia/Dockerfile
+++ b/nvidia/Dockerfile
@@ -2,6 +2,8 @@ FROM ubuntu:20.04
 
 LABEL maintainer "Jeffrey Phillips Freeman <the@jeffreyfreeman.me>"
 
+WORKDIR /tmp
+
 RUN apt-get update && \
     apt-get -y dist-upgrade && \
     rm -rf /var/lib/{cache,log}
@@ -29,6 +31,8 @@ RUN mkdir -p /etc/OpenCL/vendors && \
 ARG aparapiver
 RUN mvn org.apache.maven.plugins:maven-dependency-plugin:2.1:get -DrepoUrl="https://repo1.maven.org/maven2/" -Dartifact="com.aparapi:aparapi:${aparapiver}"
 
+RUN rm -rf /tmp/*
+
 # nvidia-container-runtime
 ENV NVIDIA_VISIBLE_DEVICES all
 ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
-- 
GitLab