py38-pytorch (python/py-pytorch) Updated: 1 year ago Add to my watchlist
Tensors and dynamic neural networks in Python with strong GPU accelerationPyTorch is a Python package that provides two high-level features: Tensor computation (like NumPy) with strong GPU acceleration; Deep neural networks built on a tape-based autograd system. You can reuse your favorite Python packages such as NumPy, SciPy and Cython to extend PyTorch when needed.
Version: 2.0.0 License: BSD
GitHub
| Maintainers | No Maintainer |
| Categories | python |
| Homepage | https://pytorch.org/ |
| Platforms | {darwin >= 19} |
| Variants |
|
Subport(s) (4)
"py38-pytorch" depends on
lib (22)
run (1)
build (10)
fetch (1)
test (1)
Ports that depend on "py38-pytorch"
lib (3)run (6)
Port notes
The port py38-pytorch is built with Apple Metal Performance Shaders (MPS) support for GPU hardware acceleration. To enable Apple GPU devices, use device "mps". Matrix multiplication example:
import torch
mpsDevice = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
x = torch.randn((10_000, 1_000), device=mpsDevice)
cov = (x.T @ x)/x.shape[0]
Port Health:
Loading Port Health