Hello, may I ask what version of pytorch and cuda you use?

[Cloud-Liu]

Hello, may I ask what version of pytorch and cuda you use?

[Heng14]

Hi Could,

I don't remember the exact version. But I think you can try CUDA 10.2 + PyTorch 1.12 to see if it works. Let me know if you have any questions.

Thanks!
Heng