The previous requirements file specified PyTorch 2.2.0 with CUDA 11.8 (cu118). However, based on the available packages from the PyTorch stable and PyTorch Geometric (PyG) URLs, PyTorch 2.2.0 is only available with CUDA 12.1 (cu121) for GPU installation.
This commit updates the requirements to use the correct CUDA version:
Update PyTorch requirement to use CUDA 12.1
The previous requirements file specified PyTorch 2.2.0 with CUDA 11.8 (cu118). However, based on the available packages from the PyTorch stable and PyTorch Geometric (PyG) URLs, PyTorch 2.2.0 is only available with CUDA 12.1 (cu121) for GPU installation.
This commit updates the requirements to use the correct CUDA version:
torch==2.2.0+cu118totorch==2.2.0+cu121