Can this code support higher versions of pytorch? I was unable to successfully configure the required pytorch environment.