Wayra-Perplexity-Estimator-55M-TensorRT / tensorrt_requirements.txt
oflorez's picture
Upload TensorRT WayraPPL model optimized for A100 GPUs
2991961 verified
raw
history blame contribute delete
633 Bytes
# TensorRT Model Requirements - A100 GPU Only
# Generated from successful environment
huggingface-hub==0.33.4
numpy @ file:///home/conda/feedstock_root/build_artifacts/numpy_1751342122905/work/dist/numpy-2.3.1-cp312-cp312-linux_x86_64.whl#sha256=a5957bb3eb8e75fda1f449c1aeeb6cea390dd20b9f44fb5a9aeea4d102158962
nvidia-cuda-runtime-cu12==12.9.79
pycuda==2025.1.1
tensorrt==10.13.0.35
tensorrt_cu12==10.13.0.35
tensorrt_cu12_bindings==10.13.0.35
tensorrt_cu12_libs==10.13.0.35
torch==2.4.0
torchaudio==2.4.0
torchvision==0.19.0
transformers==4.53.2
# Hardware: NVIDIA A100-SXM4-80GB, CUDA 12.8+, TensorRT 10.13.x, Driver 570.124.06+