From 2a41a1ec64567097973b68bb6021afd679259250 Mon Sep 17 00:00:00 2001 From: Andrew Reusch Date: Wed, 19 Oct 2022 12:01:45 -0700 Subject: [PATCH] Add python venvs to demo_ images that build. --- docker/Dockerfile.demo_android | 7 +++++++ docker/Dockerfile.demo_rocm | 8 ++++++-- docker/Dockerfile.demo_vitis_ai | 4 ++++ 3 files changed, 17 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.demo_android b/docker/Dockerfile.demo_android index 8a461269e75d..cb5a9e0015ab 100644 --- a/docker/Dockerfile.demo_android +++ b/docker/Dockerfile.demo_android @@ -28,8 +28,12 @@ RUN bash /install/ubuntu_setup_tz.sh COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh RUN bash /install/ubuntu_install_core.sh +ENV TVM_VENV /venv/apache-tvm-py3.7 +COPY python/bootstrap/lockfiles /install/python/bootstrap/lockfiles COPY install/ubuntu_install_python.sh /install/ubuntu1804_install_python.sh RUN bash /install/ubuntu1804_install_python.sh +ENV PATH ${TVM_VENV}/bin:$PATH +ENV PYTHONNOUSERSITE 1 # Disable .local directory from affecting CI. COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh RUN bash /install/ubuntu_install_python_package.sh @@ -54,6 +58,9 @@ RUN bash /install/ubuntu_install_vulkan.sh ENV VULKAN_SDK=/usr +COPY install/ubuntu_install_cmake_source.sh /install/ubuntu_install_cmake_source.sh +RUN bash /install/ubuntu_install_cmake_source.sh + RUN git clone https://github.com/KhronosGroup/OpenCL-Headers /usr/local/OpenCL-Headers/ # Build TVM diff --git a/docker/Dockerfile.demo_rocm b/docker/Dockerfile.demo_rocm index 3f3ffef78e68..b4cb83f7f68c 100644 --- a/docker/Dockerfile.demo_rocm +++ b/docker/Dockerfile.demo_rocm @@ -26,8 +26,12 @@ RUN bash /install/ubuntu_setup_tz.sh COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh RUN bash /install/ubuntu_install_core.sh -COPY install/ubuntu1804_install_python.sh /install/ubuntu1804_install_python.sh -RUN bash /install/ubuntu1804_install_python.sh +ENV TVM_VENV /venv/apache-tvm-py3.7 +COPY python/bootstrap/lockfiles /install/python/bootstrap/lockfiles +COPY install/ubuntu_install_python.sh /install/ubuntu_install_python.sh +RUN bash /install/ubuntu_install_python.sh +ENV PATH ${TVM_VENV}/bin:$PATH +ENV PYTHONNOUSERSITE 1 # Disable .local directory from affecting CI. COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh RUN bash /install/ubuntu_install_python_package.sh diff --git a/docker/Dockerfile.demo_vitis_ai b/docker/Dockerfile.demo_vitis_ai index fa024767fd1e..c90091e7ecd2 100644 --- a/docker/Dockerfile.demo_vitis_ai +++ b/docker/Dockerfile.demo_vitis_ai @@ -32,8 +32,12 @@ RUN bash /install/ubuntu_install_core.sh COPY install/ubuntu_install_vitis_ai_core.sh /install/ubuntu_install_vitis_ai_core.sh RUN bash /install/ubuntu_install_vitis_ai_core.sh +ENV TVM_VENV /venv/apache-tvm-py3.7 +COPY python/bootstrap/lockfiles /install/python/bootstrap/lockfiles COPY install/ubuntu_install_python.sh /install/ubuntu_install_python.sh RUN bash /install/ubuntu_install_python.sh +ENV PATH ${TVM_VENV}/bin:$PATH +ENV PYTHONNOUSERSITE 1 # Disable .local directory from affecting CI. COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh RUN bash /install/ubuntu_install_python_package.sh