PyTorch

Learn about PyTorch conda environments.

PyTorch 2.4.0 for GPU on Python 3.10 with CUDA 12.4 (version 1.0)
PyTorch 2.3.1 for GPU on Python 3.10 with CUDA 12.1 (version 1.0)
PyTorch 2.1 for GPU on Python 3.9 (version 1.0) with CUDA 11.8
PyTorch 2.0 for GPU on Python 3.9 (version 2.0) with CUDA 11.8