We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f6585e8 commit 7c04f11Copy full SHA for 7c04f11
setup.py
@@ -282,7 +282,8 @@ def run(self):
282
THD_LIB = os.path.join(lib_path, 'libTHD.1.dylib')
283
NCCL_LIB = os.path.join(lib_path, 'libnccl.1.dylib')
284
285
-if WITH_NCCL and subprocess.call('ldconfig -p | grep libnccl >/dev/null', shell=True) == 0:
+if WITH_NCCL and (subprocess.call('ldconfig -p | grep libnccl >/dev/null', shell=True) == 0 or
286
+ subprocess.call('/sbin/ldconfig -p | grep libnccl >/dev/null', shell=True) == 0):
287
SYSTEM_NCCL = True
288
289
main_compile_args = ['-D_THP_CORE']
0 commit comments