Update version compatility after release 0.6.0 (pytorch#1090)

Summary: Pull Request resolved: pytorch#1090

Test Plan: Imported from OSS

Reviewed By: ejguan

Differential Revision: D44107011

Pulled By: NivekT

fbshipit-source-id: 1e1efda61f9cdbfa3212a484ac2d465d744d43e9

README.md

@@ -54,6 +54,7 @@ The following is the corresponding `torchdata` versions and supported Python ver
 | `torch`              | `torchdata`        | `python`          |
 | -------------------- | ------------------ | ----------------- |
 | `master` / `nightly` | `main` / `nightly` | `>=3.8`, `<=3.11` |
+| `2.0.0`              | `0.6.0`            | `>=3.8`, `<=3.11` |
 | `1.13.1`             | `0.5.1`            | `>=3.7`, `<=3.10` |
 | `1.12.1`             | `0.4.1`            | `>=3.7`, `<=3.10` |
 | `1.12.0`             | `0.4.0`            | `>=3.7`, `<=3.10` |

setup.py

@@ -110,7 +110,7 @@ def _get_requirements():
 
 
 # Use new version of torch on main branch
-pytorch_package_dep = "torch>1.13"
+pytorch_package_dep = "torch>2.0"
 if os.getenv("PYTORCH_VERSION"):
     pytorch_package_dep = pytorch_package_dep.split(">")[0]
     pytorch_package_dep += "==" + os.getenv("PYTORCH_VERSION")