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.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 | ✓ |