1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
version: "3.4"
# To build images, run:
# docker compose -f docker-compose-aarch64.yml build
# Find MOFED_VERSION per OS and arch:
# https://network.nvidia.com/products/infiniband-drivers/linux/mlnx_ofed/
# Find NV_DRIVER_VERSION per CUDA version, OS and arch:
# 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
services:
centos8-mofed5-cuda11:
image: centos8-mofed5-cuda11:3
build:
context: .
network: host
dockerfile: centos-release.Dockerfile
args:
MOFED_VERSION: 5.4-3.6.8.1
MOFED_OS: rhel8.2
CUDA_VERSION: 11.4.0
OS_VERSION: 8
ARCH: aarch64
ubuntu18.04-mofed5-cuda11:
image: ubuntu18.04-mofed5-cuda11:3
build:
context: .
network: host
dockerfile: ubuntu-release.Dockerfile
args:
MOFED_VERSION: 5.4-3.6.8.1
UBUNTU_VERSION: 18.04
CUDA_VERSION: 11.4.0
NV_DRIVER_VERSION: 470
ARCH: aarch64
ubuntu20.04-mofed5-cuda11:
image: ubuntu20.04-mofed5-cuda11:3
build:
context: .
network: host
dockerfile: ubuntu-release.Dockerfile
args:
MOFED_VERSION: 5.4-3.6.8.1
UBUNTU_VERSION: 20.04
CUDA_VERSION: 11.4.0
NV_DRIVER_VERSION: 470
ARCH: aarch64
ubuntu22.04-mofed5-cuda11:
image: ubuntu22.04-mofed5-cuda11:3
build:
context: .
network: host
dockerfile: ubuntu-release.Dockerfile
args:
MOFED_VERSION: 5.4-3.6.8.1
UBUNTU_VERSION: 22.04
CUDA_VERSION: 11.7.0
NV_DRIVER_VERSION: 515
ARCH: aarch64
ubuntu20.04-mofed5-cuda12:
image: ubuntu20.04-mofed5-cuda12:3
build:
context: .
network: host
dockerfile: ubuntu-release.Dockerfile
args:
MOFED_VERSION: 5.4-3.6.8.1
UBUNTU_VERSION: 20.04
CUDA_VERSION: 12.0.0
NV_DRIVER_VERSION: 525
ARCH: aarch64
ubuntu22.04-mofed5-cuda12:
image: ubuntu22.04-mofed5-cuda12:3
build:
context: .
network: host
dockerfile: ubuntu-release.Dockerfile
args:
MOFED_VERSION: 5.4-3.6.8.1
UBUNTU_VERSION: 22.04
CUDA_VERSION: 12.0.0
NV_DRIVER_VERSION: 525
ARCH: aarch64
|
