"...include/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "3d264140c31fa90fcf4f8a3aeaa9108d0afe56c9"
README.md 843 Bytes
Newer Older
1
2
3
To ease the process of installing all the dependencies, we provide a Dockerfile and a simple guideline to build a Docker image with all of above installed. The Docker image is built on top of Ubuntu 20.04, and it contains all the dependencies required to run the experiments. We only provide the Dockerfile for NVIDIA GPU, and the Dockerfile for AMD GPU will be provided upon request.

```bash
4
git clone --recursive https://github.com/tile-ai/tilelang TileLang
5
6
7
8
9
10
cd TileLang/docker
# build the image, this may take a while (around 10+ minutes on our test machine)
docker build -t tilelang_cuda -f Dockerfile.cu120 .
# run the container
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
```