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 | NIC5 | Hercules2 | Dragon2 | Lemaitre3 | Dragon1 |
|---|---|---|---|---|---|
| PyTorch/1.10.0-foss-2021a | ✓ | ||||
| PyTorch/1.10.0-foss-2021a-CUDA-11.3.1 | ✓ | ||||
| PyTorch/1.12.0-foss-2022a | ✓ | ||||
| PyTorch/1.12.0-foss-2022a-CUDA-11.7.0 | ✓ | ✓ | |||
| PyTorch/1.12.1-foss-2021b | ✓ | ||||
| PyTorch/1.12.1-foss-2022a-CUDA-11.7.0 | ✓ | ||||
| PyTorch/1.4.0-foss-2019b-Python-3.7.4 | ✓ | ✓ | |||
| PyTorch/1.4.0-fosscuda-2019b-Python-3.7.4 | ✓ | ||||
| PyTorch/1.6.0-foss-2019b-Python-3.7.4 | ✓ | ||||
| PyTorch/1.6.0-fosscuda-2019b-Python-3.7.4 | ✓ | ||||
| PyTorch/1.9.0-foss-2020b | ✓ | ||||
| PyTorch/1.9.0-fosscuda-2020b | ✓ | ||||
| PyTorch/2.1.2-foss-2022b | ✓ | ||||
| PyTorch/2.1.2-foss-2023b | ✓ |