The only way I found to make it work is to use a docker container with the required PyTorch version in it.