What version of PyTorch should we use with Ray Train?

Ray Libraries (Data, Train, Tune, Serve) Ray Train
1

What version of PyTorch should we use with Ray Train? (ray 1.9.1)

2

There isn’t a requirement, but ray-ml docker images are built with 1.9.0 (or 1.9.0+cu111 for GPU) and have been tested to work!