NetArena/Dockerfile at a2a-agentx · Froot-NetSys/NetArena · GitHub

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
84
FROM nvidia/cuda:12.2.2-devel-ubuntu22.04

# Need Wget to install Miniconda.
RUN  apt-get update && apt-get install -y wget

# Install Miniconda.
RUN mkdir -p /miniconda3
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /miniconda3/miniconda.sh
RUN bash /miniconda3/miniconda.sh -b -u -p /miniconda3
RUN rm /miniconda3/miniconda.sh

ENV PATH="/miniconda3/bin:${PATH}"

WORKDIR /NetArena

# Build dependencies.
RUN apt-get install -y build-essential

# Create conda environments.
RUN conda tos accept
COPY --chown=user environment_ai_gym.yml .
RUN conda env create -f environment_ai_gym.yml
COPY --chown=user environment_mininet.yml .
RUN conda env create -f environment_mininet.yml

# Install Flash Attention separately since it requires PyTorch at build time...
RUN conda init --all
RUN conda run -n ai_gym_env pip install flash-attn==2.7.4.post1

# Make sudo dummy replacement that handles env vars properly.
RUN echo '#!/bin/bash\n\
if [ "$1" = "-E" ]; then shift; fi\n\
while [[ "$1" == *=* ]]; do export "$1"; shift; done\n\
"$@"' > /usr/bin/sudo && chmod +x /usr/bin/sudo

# Install Mininet and dependencies.
RUN apt-get update && apt-get install -y lsb-release git
RUN git clone https://github.com/mininet/mininet /opt/mininet
RUN cd /opt/mininet/util && DEBIAN_FRONTEND=noninteractive ./install.sh -a

# Kubenertes CLI.
RUN apt-get install -y curl ca-certificates gnupg2
RUN curl -LO "https://dl.k8s.io/release/v1.34.0/bin/linux/amd64/kubectl"
RUN chmod +x ./kubectl
RUN mv ./kubectl /usr/local/bin/kubectl

# Add Docker's official GPG key:
RUN install -m 0755 -d /etc/apt/keyrings
RUN curl -fsSL https://download.docker.com/linux/ubuntu/gpg -o /etc/apt/keyrings/docker.asc
RUN chmod a+r /etc/apt/keyrings/docker.asc

# Add the repository to Apt sources:
RUN echo \
  "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.asc] https://download.docker.com/linux/ubuntu \
  $(. /etc/os-release && echo "${UBUNTU_CODENAME:-$VERSION_CODENAME}") stable" | \
  sudo tee /etc/apt/sources.list.d/docker.list > /dev/null

# Install Docker CLI.
RUN apt-get update && apt-get install -y docker-ce-cli

# Install KIND.
RUN apt-get install -y golang-go
RUN go install sigs.k8s.io/kind@v0.26.0
RUN mv /root/go/ /usr/local/
ENV GOPATH=/usr/local/go
ENV PATH=$PATH:$GOPATH/bin

# Skaffold.
RUN curl -Lo skaffold https://storage.googleapis.com/skaffold/releases/v2.15.0/skaffold-linux-amd64
RUN chmod +x skaffold
RUN mv skaffold /usr/local/bin

# Fetch Microservices repository to /microservices-demo (outside /NetArena so volume mounts don't hide it)
RUN apt-get install -y git
RUN git clone https://github.com/GoogleCloudPlatform/microservices-demo.git /microservices-demo
RUN sed -i 's|$BUILDPLATFORM|linux/amd64|g' /microservices-demo/src/loadgenerator/Dockerfile

# Install network utilities for Mininet (ping, traceroute, etc.)
RUN apt-get update && apt-get install -y iputils-ping net-tools iproute2 traceroute

# Clean up apt cache to reduce image size.
RUN rm -rf /var/lib/apt/lists/*

COPY --chown=user . .