TensorRT-LLMs/docker/common/install_polygraphy.sh
2023-10-10 23:22:17 -07:00

8 lines
225 B
Bash

#!/bin/bash
set -ex
RELEASE_URL_PG=https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/secure/9.0.1/tars/polygraphy-0.48.1-py2.py3-none-any.whl
pip uninstall -y polygraphy
pip install ${RELEASE_URL_PG}