Skip to content

Commit 9285e38

Browse files
LeiWang1999LeiWang1999
authored
Add Dockerfiles for multiple CUDA versions (#103)
Create Dockerfiles for CUDA versions 11.8, 12.1, 12.3, 12.4, and 12.5, using the latest PyTorch NGC base images. Update docker/README.md to provide clearer instructions for building and running the Docker container with a specific CUDA version.
1 parent 778dbd2 commit 9285e38

File tree

7 files changed

+170
-1
lines changed

7 files changed

+170
-1
lines changed

‎docker/Dockerfile.cu118‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
FROM nvcr.io/nvidia/pytorch:22.12-py3
2+
3+
WORKDIR /root
4+
5+
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
6+
7+
ENV DEBIAN_FRONTEND=noninteractive
8+
9+
RUN apt-get update && apt-get install -y --no-install-recommends \
10+
build-essential git wget \
11+
libgtest-dev libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev llvm-dev \
12+
&& apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
13+
14+
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0-Linux-x86_64.sh -O install_miniconda.sh && \
15+
bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
16+
17+
ENV PATH="/opt/conda/bin:${PATH}"
18+
19+
ENV LIBGL_ALWAYS_INDIRECT=1
20+
21+
RUN conda install pip cmake && conda clean --all
22+
23+
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
24+
25+
RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main TileLang \
26+
&& cd TileLang && ./install_cuda.sh
27+
28+
CMD bash

‎docker/Dockerfile.cu121‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
FROM nvcr.io/nvidia/pytorch:23.07-py3
2+
3+
WORKDIR /root
4+
5+
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
6+
7+
ENV DEBIAN_FRONTEND=noninteractive
8+
9+
RUN apt-get update && apt-get install -y --no-install-recommends \
10+
build-essential git wget \
11+
libgtest-dev libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev llvm-dev \
12+
&& apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
13+
14+
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0-Linux-x86_64.sh -O install_miniconda.sh && \
15+
bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
16+
17+
ENV PATH="/opt/conda/bin:${PATH}"
18+
19+
ENV LIBGL_ALWAYS_INDIRECT=1
20+
21+
RUN conda install pip cmake && conda clean --all
22+
23+
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
24+
25+
RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main TileLang \
26+
&& cd TileLang && ./install_cuda.sh
27+
28+
CMD bash

‎docker/Dockerfile.cu123‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
FROM nvcr.io/nvidia/pytorch:24.02-py3
2+
3+
WORKDIR /root
4+
5+
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
6+
7+
ENV DEBIAN_FRONTEND=noninteractive
8+
9+
RUN apt-get update && apt-get install -y --no-install-recommends \
10+
build-essential git wget \
11+
libgtest-dev libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev llvm-dev \
12+
&& apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
13+
14+
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0-Linux-x86_64.sh -O install_miniconda.sh && \
15+
bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
16+
17+
ENV PATH="/opt/conda/bin:${PATH}"
18+
19+
ENV LIBGL_ALWAYS_INDIRECT=1
20+
21+
RUN conda install pip cmake && conda clean --all
22+
23+
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
24+
25+
RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main TileLang \
26+
&& cd TileLang && ./install_cuda.sh
27+
28+
CMD bash

‎docker/Dockerfile.cu124‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
FROM nvcr.io/nvidia/pytorch:24.05-py3
2+
3+
WORKDIR /root
4+
5+
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
6+
7+
ENV DEBIAN_FRONTEND=noninteractive
8+
9+
RUN apt-get update && apt-get install -y --no-install-recommends \
10+
build-essential git wget \
11+
libgtest-dev libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev llvm-dev \
12+
&& apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
13+
14+
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0-Linux-x86_64.sh -O install_miniconda.sh && \
15+
bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
16+
17+
ENV PATH="/opt/conda/bin:${PATH}"
18+
19+
ENV LIBGL_ALWAYS_INDIRECT=1
20+
21+
RUN conda install pip cmake && conda clean --all
22+
23+
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
24+
25+
RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main TileLang \
26+
&& cd TileLang && ./install_cuda.sh
27+
28+
CMD bash

‎docker/Dockerfile.cu125‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
FROM nvcr.io/nvidia/pytorch:24.07-py3
2+
3+
WORKDIR /root
4+
5+
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
6+
7+
ENV DEBIAN_FRONTEND=noninteractive
8+
9+
RUN apt-get update && apt-get install -y --no-install-recommends \
10+
build-essential git wget \
11+
libgtest-dev libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev llvm-dev \
12+
&& apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
13+
14+
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0-Linux-x86_64.sh -O install_miniconda.sh && \
15+
bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
16+
17+
ENV PATH="/opt/conda/bin:${PATH}"
18+
19+
ENV LIBGL_ALWAYS_INDIRECT=1
20+
21+
RUN conda install pip cmake && conda clean --all
22+
23+
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
24+
25+
RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main TileLang \
26+
&& cd TileLang && ./install_cuda.sh
27+
28+
CMD bash

‎docker/Dockerfile.cu126‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
FROM nvcr.io/nvidia/pytorch:24.12-py3
2+
3+
WORKDIR /root
4+
5+
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
6+
7+
ENV DEBIAN_FRONTEND=noninteractive
8+
9+
RUN apt-get update && apt-get install -y --no-install-recommends \
10+
build-essential git wget \
11+
libgtest-dev libprotobuf-dev protobuf-compiler libgflags-dev libsqlite3-dev llvm-dev \
12+
&& apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
13+
14+
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.5.2-0-Linux-x86_64.sh -O install_miniconda.sh && \
15+
bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
16+
17+
ENV PATH="/opt/conda/bin:${PATH}"
18+
19+
ENV LIBGL_ALWAYS_INDIRECT=1
20+
21+
RUN conda install pip cmake && conda clean --all
22+
23+
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
24+
25+
RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main TileLang \
26+
&& cd TileLang && ./install_cuda.sh
27+
28+
CMD bash

‎docker/README.md‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@ To ease the process of installing all the dependencies, we provide a Dockerfile
44
git clone --recursive https://github.com/tile-ai/tilelang TileLang
55
cd TileLang/docker
66
# build the image, this may take a while (around 10+ minutes on our test machine)
7-
docker build -t tilelang_cuda -f Dockerfile.cu120 .
7+
# replace the version number cu124 with the one you want to use
8+
docker build -t tilelang_cuda -f Dockerfile.cu124 .
89
# run the container
910
docker run -it --cap-add=SYS_ADMIN --network=host --gpus all --cap-add=SYS_PTRACE --shm-size=4G --security-opt seccomp=unconfined --security-opt apparmor=unconfined --name tilelang_test tilelang_cuda bash
1011
```

0 commit comments

Comments
 (0)