Tencent / TPAT

TensorRT Plugin Autogen Tool
Apache License 2.0
367 stars 42 forks source link

unsupported ptx version error #32

Closed frankxyy closed 1 year ago

frankxyy commented 1 year ago

image

I run the example for onehot plugin and ran into this error. The gpu info printed out with lspci | grep -i vga command is : 1VGA compatible controller: NVIDIA Corporation Device 2204 (rev a1)1. I think this is rtx3090 gpu. The nvidia driver version 510.

However, on another machine with lspci | grep -i vga command printed out: 37:00.0 VGA compatible controller: NVIDIA Corporation GA102 [GeForce RTX 3090] (rev a1), the onehot example can be run successfully (the nvidia driver version is 515). This machine seems to be also with rtx3090 gpu but the lspci | grep -i vga command printed out message is different from the previous machine. Right now I am not able to find out the cause.