aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix install nccl setup.py

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-24 19:48:37 -03:00
parent b637c3e0f1
commit ef272b65ee
44 changed files with 7650 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
setup.py
View file

@ -18,7 +18,7 @@ apt_get_dependencies = [
apt_nccl = [
'libnccl2=2.21.5-1+cuda12.2',
'libnccl-dev=2.21.5-1+cuda12.2 '
'libnccl-dev=2.21.5-1+cuda12.2'
]
subprocess.check_call(['sudo', 'apt-get', 'update'])
@ -27,14 +27,16 @@ for package in apt_dependencies:
subprocess.check_call(['sudo', 'apt', 'install', package])
for package in apt_get_dependencies:
subprocess.check_call(['sudo', 'apt-get', 'install', 'y', package])
subprocess.check_call(['sudo', 'apt-get', 'install', '-y', package])
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(['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'])
for package in apt_nccl:
subprocess.check_call(['sudo', 'apt', 'install', package])
subprocess.check_call(['rm', 'cuda-keyring_1.0-1_all.deb'])
class CustomInstall(install):
def run(self):
subprocess.call(['make', '-C', 'build'])