diff --git a/amdgpu/Dockerfile b/amdgpu/Dockerfile
index 6d3242e441d6b44c2bd22449a1e0b60439703cda..7088050fae884770b6ac66f935f5d6a9a150dd3f 100644
--- a/amdgpu/Dockerfile
+++ b/amdgpu/Dockerfile
@@ -39,6 +39,8 @@ 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}
 
+ENV JAVA_HOME /usr/lib/jvm/java-14-openjdk-amd64/
+
 #RUN dpkg --add-architecture i386 && \
 #    apt-get update && apt-get -y dist-upgrade  && \
 #    apt-get -y --no-install-recommends install ca-certificates curl xz-utils
diff --git a/nvidia/Dockerfile b/nvidia/Dockerfile
index b6038bca4b4fab7b51aa3678e276d882f7d9d5fc..124dd40603fe032a7fa66a4c39637dc49159cf59 100644
--- a/nvidia/Dockerfile
+++ b/nvidia/Dockerfile
@@ -32,3 +32,4 @@ RUN mvn org.apache.maven.plugins:maven-dependency-plugin:2.1:get -DrepoUrl="http
 # nvidia-container-runtime
 ENV NVIDIA_VISIBLE_DEVICES all
 ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
+ENV JAVA_HOME /usr/lib/jvm/java-14-openjdk-amd64/