pytorch: Open-source machine learning library
Contents
Overview of package
Package: | pytorch |
---|---|
Description: | Open-source machine learning library |
For more information: | https://pytorch.org/ |
Categories: | |
License: | OpenSource (BSD) |
General usage information
PyTorch is an open-source machine learning library that evolved from the (no longer supported) Lua based Torch library. It is commonly used for computer vision and natural language processing tasks. Among its high level features are:
- Tensor computing (like NumPy) with strong GPU acceleration
- Deep neural networks built on a tape-based automatic differentiation system
The torch module has been added to your PYTHONPATH
Available versions of the package pytorch, by cluster
This section lists the available versions of the package pytorchon the different clusters.
Available versions of pytorch on the Deepthought2 cluster (RHEL8)
Version | Module tags | CPU optimized for | GPU ready? |
---|---|---|---|
1.5.0 | pytorch/1.5.0 | ivybridge | Y |