From 8654374a334064ae42e23726fe76980f47e66fd1 Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Fri, 24 May 2024 20:13:16 -0300 Subject: [PATCH] fix setup.py install nccl --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index e8d6c28..083368d 100644 --- a/setup.py +++ b/setup.py @@ -32,6 +32,7 @@ for package in apt_get_dependencies: subprocess.check_call(['wget', 'https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb']) subprocess.check_call(['sudo', 'dpkg', '-i', 'cuda-keyring_1.0-1_all.deb']) +subprocess.check_call(['sudo', 'apt-get', 'upgrade', '-y', '--allow-change-held-packages']) subprocess.check_call(['sudo', 'apt-mark', 'unhold', 'libnccl-dev', 'libnccl2']) for package in apt_nccl: