From 3c75538c82c8b146d63ea501dffc3d1932210088 Mon Sep 17 00:00:00 2001 From: Tongliang Liao Date: Fri, 6 Nov 2020 20:05:36 +0800 Subject: [PATCH] Use rel symlink for CUDA due to nvidia docker issue: https://github.com/NVIDIA/libnvidia-container/issues/117 --- pkgs/cuda.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pkgs/cuda.sh b/pkgs/cuda.sh index 142a6c7a..76330150 100644 --- a/pkgs/cuda.sh +++ b/pkgs/cuda.sh @@ -49,7 +49,9 @@ # - cuda-toolkit creates symlink "/usr/local/cuda -> /etc/alternatives/cuda -> /usr/local/cuda-". # libnvidia-container cannot set up cuda-compat driver properly in this case. # Use "ln -T" to overwrite dir symlink. - ls -d "/usr/local/cuda-$CUDA_VER_MAJOR.$CUDA_VER_MINOR/" | sort -V | tail -n1 | sudo xargs -I{} ln -sfT {} '/usr/local/cuda' + # - Only rel path "/usr/local/cuda -> cuda-" works for libnvidia-container, not "/usr/local/cuda -> /usr/local/cuda-". + # https://github.com/NVIDIA/libnvidia-container/issues/117 + ls -d "/usr/local/cuda-$CUDA_VER_MAJOR.$CUDA_VER_MINOR/" | sort -V | tail -n1 | xargs -n1 basename | sudo xargs -I{} ln -sfT {} "/usr/local/cuda" else case "$DISTRO_ID" in 'centos' | 'fedora' | 'rhel')