"examples/model_compress/pruning/admm_pruning_torch.py" did not exist on "f46f0cf4b7013dac1a57ae9f28a1e00cc077a71b"
docker_pypi_distribute.sh 469 Bytes
Newer Older
1
# Get the CUDA version from the command line
2
3
IMAGE="tilelang-builder:manylinux"
docker build . -f "$(dirname "${BASH_SOURCE[0]}")/pypi.manylinux.Dockerfile" --tag ${IMAGE}
4
5
6

install_pip="python3.8 -m pip install --upgrade pip && python3.8 -m pip install -r requirements-build.txt"

7
tox_command="python3.8 -m tox -e py38-pypi,py39-pypi,py310-pypi,py311-pypi,py312-pypi"
8

9
docker run --rm --gpus all -v $(pwd):/tilelang ${IMAGE} /bin/bash -c "$install_pip && $tox_command"