Required python and CUDA versions

Hi,

I've been trying to get your model up and running on my systems. I've been having some trouble with the listed requirements, specifically, I was wondering if you could specify what versions for the following you used?

- python
- pytorch
- cuda

Thank you!