This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new 39b2db4 [DOCKER] Fix vulkansdk in the ci-gpu (#5566)
39b2db4 is described below
commit 39b2db4bd88266c25b203b18571f960d0cc64896
Author: Tianqi Chen <[email protected]>
AuthorDate: Mon May 11 15:28:04 2020 -0700
[DOCKER] Fix vulkansdk in the ci-gpu (#5566)
---
docker/Dockerfile.ci_gpu | 5 +----
docker/install/ubuntu_install_vulkan.sh | 11 ++++-------
2 files changed, 5 insertions(+), 11 deletions(-)
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
index 1625496..d829014 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -111,7 +111,4 @@ ENV
LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/compact:${LD_LIBRARY_P
ENV LD_LIBRARY_PATH=/opt/rocm/lib:${LD_LIBRARY_PATH}
ENV PATH=/node_modules/.bin:${PATH}
-ENV VULKAN_SDK=/usr/local/VulkanSDK/1.0.65.0/x86_64
-ENV PATH=${PATH}:${VULKAN_SDK}/bin
-ENV LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${VULKAN_SDK}/lib
-ENV VK_LAYER_PATH=${VULKAN_SDK}/etc/explicit_layer.d
+ENV VULKAN_SDK=/usr
diff --git a/docker/install/ubuntu_install_vulkan.sh
b/docker/install/ubuntu_install_vulkan.sh
index 5fb4082..b7d2d46 100755
--- a/docker/install/ubuntu_install_vulkan.sh
+++ b/docker/install/ubuntu_install_vulkan.sh
@@ -20,10 +20,7 @@ set -e
set -u
set -o pipefail
-wget -q
https://sdk.lunarg.com/sdk/download/1.0.65.0/linux/vulkansdk-linux-x86_64-1.0.65.0.run
-
-bash vulkansdk-linux-x86_64-1.0.65.0.run
-mv VulkanSDK /usr/local/VulkanSDK
-cd /usr/local/VulkanSDK/1.0.65.0
-./build_tools.sh
-./build_samples.sh
+wget -qO - http://packages.lunarg.com/lunarg-signing-key-pub.asc | apt-key add
-
+wget -qO /etc/apt/sources.list.d/lunarg-vulkan-1.2.135-xenial.list
http://packages.lunarg.com/vulkan/1.2.135/lunarg-vulkan-1.2.135-xenial.list
+apt update
+apt install -y vulkan-sdk
