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

AI & Data Science Deep Learning (Training & Inference) TensorRT
, , ,
1

Description

When I try to install the tensorRT using tar file, I come across this ERROR when trying to install the related python wheel: tensorrt-8.2.0.6-cp38-none-linux_x86_64.whl 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!

Environment

TensorRT Version: 8.2.0.6
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

2

Hi,

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

image
image2422×1792 419 KB

Thank you.

3

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.

4

I have solved the problem by trying tensorrt-8.2.0.6-cp39-none-linux_x86_64.whl. Somehow weird because my python version is 3.8.0.

5

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