Spaces:
Paused
Paused
File size: 500 Bytes
3ec0fd4 25adfbc 3ec0fd4 25adfbc 3ec0fd4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/bin/bash -e
export ENV=${ENV:-/etc/shinit_v2}
source $ENV
CUDA_ARCH="${CUDA_ARCH:-89-real;90-real}"
cd /root/TensorRT-LLM
python3 scripts/build_wheel.py --clean --cuda_architectures "$CUDA_ARCH" --trt_root /usr/local/tensorrt
pip install build/tensorrt_llm-0.7.1-cp310-cp310-linux_x86_64.whl
mv examples ../TensorRT-LLM-examples
cd ..
rm -rf TensorRT-LLM
# we don't need static libraries and they take a lot of space
(cd /usr && find . -name '*static.a' | grep -v cudart_static | xargs rm -f)
|