TDA4VM: QAT recommended GPU driver version

Ming Zhong10

Part Number: TDA4VM

Dear expert,

Our customer found that QAT is not compatible with their Nvidia driver. What is recommended driver version? Pytorch version?

Thanks & Best Regards!

0 Manu Mathew over 5 years ago

Hi ZM,

The QAT implementation that we have is just a thin wrapper over Pytorch Modules. So if Pytorch works, the QAT should also work.

You can get a suitable version of Pytorch from the following link: https://pytorch.org/

For example,

It is important to install the correct version by using this link.

I am currently using Anaconda Python distribution, Python 3.7, nvidia driver 440, cuda10.2 and Pytorch version 1.5.

But I have used a variety of versions before.

Best regards,

0 Ming Zhong10 over 5 years ago in reply to Manu Mathew

Thanks Manu. I will ask them to follow this validated routine.

Processors forum