What is the feature?
Pytorch 2.1 is already a very old version, the corresponding CUDA version is not popular for recent GPU cards and Ubuntu systems. I think it is better to upgrade the library to support Pytorch 2.5 and later version.
Any other context?
No response