11version : " 3.4"
2+ # To build images, run:
3+ # docker compose -f docker-compose.yml build
24
3- # Find driver version based on CUDA version, OS and CPU arch (515 in this case):
5+ # Find MOFED_VERSION per OS and arch:
6+ # https://network.nvidia.com/products/infiniband-drivers/linux/mlnx_ofed/
7+
8+ # Find NV_DRIVER_VERSION per CUDA version, OS and arch:
49# https://developer.nvidia.com/cuda-11-7-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local
510
611services :
712 centos7-mofed5-cuda11 :
8- image : centos7-mofed5-cuda11:2
13+ image : centos7-mofed5-cuda11:3
914 build :
1015 context : .
1116 network : host
@@ -15,19 +20,9 @@ services:
1520 MOFED_OS : rhel7.6
1621 CUDA_VERSION : 11.4.0
1722 OS_VERSION : 7
18- centos7-mofed5.4-cuda11 :
19- image : centos7-mofed5.4-cuda11
20- build :
21- context : .
22- network : host
23- dockerfile : centos-release.Dockerfile
24- args :
25- MOFED_VERSION : 5.4-1.0.3.0
26- MOFED_OS : rhel7.6
27- CUDA_VERSION : 11.2.0
28- OS_VERSION : 7
23+ ARCH : x86_64
2924 centos8-mofed5-cuda11 :
30- image : centos8-mofed5-cuda11:2
25+ image : centos8-mofed5-cuda11:3
3126 build :
3227 context : .
3328 network : host
@@ -37,6 +32,7 @@ services:
3732 MOFED_OS : rhel8.2
3833 CUDA_VERSION : 11.4.0
3934 OS_VERSION : 8
35+ ARCH : x86_64
4036 ubuntu18.04-mofed5-cuda11 :
4137 image : ubuntu18.04-mofed5-cuda11:3
4238 build :
@@ -48,6 +44,7 @@ services:
4844 UBUNTU_VERSION : 18.04
4945 CUDA_VERSION : 11.4.0
5046 NV_DRIVER_VERSION : 470
47+ ARCH : x86_64
5148 ubuntu20.04-mofed5-cuda11 :
5249 image : ubuntu20.04-mofed5-cuda11:3
5350 build :
@@ -59,6 +56,7 @@ services:
5956 UBUNTU_VERSION : 20.04
6057 CUDA_VERSION : 11.4.0
6158 NV_DRIVER_VERSION : 470
59+ ARCH : x86_64
6260 ubuntu22.04-mofed5-cuda11 :
6361 image : ubuntu22.04-mofed5-cuda11:3
6462 build :
@@ -70,3 +68,52 @@ services:
7068 UBUNTU_VERSION : 22.04
7169 CUDA_VERSION : 11.7.0
7270 NV_DRIVER_VERSION : 515
71+ ARCH : x86_64
72+ centos7-mofed5-cuda12 :
73+ image : centos7-mofed5-cuda12:3
74+ build :
75+ context : .
76+ network : host
77+ dockerfile : centos-release.Dockerfile
78+ args :
79+ MOFED_VERSION : 5.0-1.0.0.0
80+ MOFED_OS : rhel7.6
81+ CUDA_VERSION : 12.0.0
82+ OS_VERSION : 7
83+ ARCH : x86_64
84+ ubuntu18.04-mofed5-cuda12 :
85+ image : ubuntu18.04-mofed5-cuda12:3
86+ build :
87+ context : .
88+ network : host
89+ dockerfile : ubuntu-release.Dockerfile
90+ args :
91+ MOFED_VERSION : 5.0-1.0.0.0
92+ UBUNTU_VERSION : 18.04
93+ CUDA_VERSION : 12.0.0
94+ NV_DRIVER_VERSION : 525
95+ ARCH : x86_64
96+ ubuntu20.04-mofed5-cuda12 :
97+ image : ubuntu20.04-mofed5-cuda12:3
98+ build :
99+ context : .
100+ network : host
101+ dockerfile : ubuntu-release.Dockerfile
102+ args :
103+ MOFED_VERSION : 5.0-1.0.0.0
104+ UBUNTU_VERSION : 20.04
105+ CUDA_VERSION : 12.0.0
106+ NV_DRIVER_VERSION : 525
107+ ARCH : x86_64
108+ ubuntu22.04-mofed5-cuda12 :
109+ image : ubuntu22.04-mofed5-cuda12:3
110+ build :
111+ context : .
112+ network : host
113+ dockerfile : ubuntu-release.Dockerfile
114+ args :
115+ MOFED_VERSION : 5.4-3.6.8.1
116+ UBUNTU_VERSION : 22.04
117+ CUDA_VERSION : 12.0.0
118+ NV_DRIVER_VERSION : 525
119+ ARCH : x86_64
0 commit comments