PyTorch¶
Description¶
Tensors and Dynamic neural networks in Python with strong GPU acceleration. PyTorch is a deep learning framework that puts Python first.
More information¶
- Homepage: https://pytorch.org/
Availability¶
| Module | Manneback | Lemaitre4 |
|---|---|---|
| PyTorch/1.10.0-fosscuda-2020b | √ | |
| PyTorch/1.11.0-foss-2021a-CUDA-11.3.1 | √ | |
| PyTorch/1.9.0-fosscuda-2020b | √ | |
| PyTorch/2.0.1-foss-2022a | √ | |
| PyTorch/2.1.2-foss-2023a-CUDA-12.1.1 | √ | |
| PyTorch/2.1.2-foss-2023b | √ |