I wrote code using the latest versions, but it doesn't work under paperspace's pytorch machines.