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

implemented mpi for distributed run

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-24 15:12:53 -03:00
parent b152ae2a1d
commit e55c1195ec
15 changed files with 105 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

8
setup.py
View file

@ -9,9 +9,17 @@ apt_dependencies = [
'nvidia-cuda-toolkit'
]
apt_get_dependencies = [
'mpi',
'libopenmpi-dev'
]
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])
class CustomInstall(install):
def run(self):
subprocess.call(['make', '-C', 'build'])