NCCL v2.10.3

docker.py

@@ -72,6 +72,7 @@
     'nccl2.7-cuda101', 'nccl2.7-cuda102', 'nccl2.7-cuda110', 'nccl2.7-cuda111',
     'nccl2.8-cuda112',
     'nccl2.9-cuda102', 'nccl2.9-cuda110', 'nccl2.9-cuda113',
+    'nccl2.10-cuda102', 'nccl2.10-cuda110',
 ]
 cutensor_choices = [
     'none',
@@ -116,8 +117,9 @@
     'cuda101': ['nccl2.4-cuda101', 'nccl2.5-cuda101', 'nccl2.6-cuda101',
                 'nccl2.7-cuda101'],
     'cuda102': ['nccl2.5-cuda102', 'nccl2.6-cuda102', 'nccl2.7-cuda102',
-                'nccl2.9-cuda102'],
-    'cuda110': ['nccl2.7-cuda110', 'nccl2.9-cuda110'],
+                'nccl2.9-cuda102', 'nccl2.10-cuda102'],
+    'cuda110': ['nccl2.7-cuda110', 'nccl2.9-cuda110', 'nccl2.10-cuda110',
+                'nccl2.10-cuda110'],
     'cuda111': ['nccl2.7-cuda111'],
     'cuda112': ['nccl2.8-cuda112'],
     'cuda113': ['nccl2.9-cuda113'],
@@ -964,6 +966,24 @@ def get_scipy_choices():
     lib_dir='/usr/lib/x86_64-linux-gnu',
 )
 
+codes['nccl2.9-cuda102'] = nccl_base.format(
+    nccl_sub_dir='cuda',
+    nccl_os_ver='1804',
+    libnccl2='libnccl2_2.10.3-1+cuda10.2_amd64',
+    libnccl_dev='libnccl-dev_2.10.3-1+cuda10.2_amd64',
+    include_dir='/usr/include',
+    lib_dir='/usr/lib/x86_64-linux-gnu',
+)
+
+codes['nccl2.9-cuda110'] = nccl_base.format(
+    nccl_sub_dir='cuda',
+    nccl_os_ver='1804',
+    libnccl2='libnccl2_2.10.3-1+cuda11.0_amd64',
+    libnccl_dev='libnccl-dev_2.10.3-1+cuda11.0_amd64',
+    include_dir='/usr/include',
+    lib_dir='/usr/lib/x86_64-linux-gnu',
+)
+
 # cuTENSOR
 # The shell script needs to be saved in an env var due to Dockerfile limitations
 codes['cutensor1.2.0-cuda101'] = 'RUN eval $CUTENSOR_INSTALL && install_cutensor 1.2.0.2 1.2.0 10.1;'