Install the right torch version #35

Job	Run time
tests	2m 6s
coverage	2m 22s
env (windows-latest)	2m 2s
env (macos-latest)	1m 55s
env (ubuntu-latest)	1m 55s
workflow-coverage / dynamo (float32)	11m 38s
workflow-coverage / overall (float32)	2m 20s
workflow-coverage / slow (float32)	6m 43s
workflow-coverage / overall (float64)	2m 35s
workflow-coverage / slow (float64)	8m 30s
workflow-docs / docstests & sphinx-build - python-3.11, torch-2.1.2	3m 23s
workflow-mypy / python-3.11, torch-2.1.2	1m 16s
workflow-tests / python-3.10, torch-2.0.0	2m 3s
workflow-tests / python-3.10, torch-2.1.2	2m 3s
workflow-tests / python-3.11, torch-2.0.0	2m 4s
workflow-tests / python-3.11, torch-2.1.2	2m 6s
workflow-coverage / report	25s
collector	1s
	55m 27s

Provide feedback