Hi. I have been trying to work with this repository for the past 3 days, and the debugging is endless. Fixing one error based on suggestions keeps leading me to a new error! Can someone list the packages' versions that are known to work with the current update of the code without any problems? Particularly python, pytorch and cuda. Thanks.