Can tensorRT python wheel be installed on the platform with amd chip using tar file?


When I try to install the tensorRT using tar file, I come across this ERROR when trying to install the related python wheel: tensorrt- is not a supported wheel on this platform. Since this wheel is for x86_64 and my cpu type is amd, I wonder if there is tar file installation method for amd64 architecture? Thank you!


TensorRT Version:
GPU Type: Nvidia A100
CPU Type: AMD EPYC 7742 64-Core Processor
Nvidia Driver Version: 450.156.00
CUDA Version: 11.0
Operating System + Version: Ubuntu 20.04


Could you please share with us uname -a output.
While downloading please choose the correct Tar file.

Thank you.

Thank you for your quick reply! I choose the correct cuda version, which is 11.0. The ‘uname -a’ result is x86_64 GNU/Linux. So I am a little bit confused whether the error is caused by the architecture.

I have solved the problem by trying tensorrt- Somehow weird because my python version is 3.8.0.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.