1. /
  2. channels
  3. /
  4. conda-forge
  5. /
  6. pytorch-gpu
conda-forge
conda-forge
public
30.9k
A community led collection of recipes, build infrastructure and distributions for the conda package manager.
Package icon
pytorch-gpu
v2.8.0
PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.

Install with Pixi

pixi add pytorch-gpu

Install with conda

conda install -c https://prefix.dev/conda-forge pytorch-gpu

Description

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

Download counts

Loading chart data...

Version

2.8.0

Platforms

linux-64
linux-aarch64
win-64

Last Published

17 days ago

Homepage

Documentation

Open Pull Requests

  •  
  •  
  •  
Package Variants
Version
Build
Created
Size
Architecture
Downloads
pytorch-gpu-2.8.0-cuda128_mkl_h2fd0c33_300.conda
2.8.0
cuda128_mkl_h2fd0c33_300 (300)
17 days ago
46.9 KB
win-64
N/A
pytorch-gpu-2.8.0-cuda129_mkl_h43a4b0b_300.conda
2.8.0
cuda129_mkl_h43a4b0b_300 (300)
18 days ago
46.31 KB
linux-64
N/A
pytorch-gpu-2.8.0-cuda129_generic_hbb4297b_200.conda
2.8.0
cuda129_generic_hbb4297b_200 (200)
18 days ago
46.35 KB
linux-64
N/A
pytorch-gpu-2.8.0-cuda129_generic_h8ef9432_200.conda
2.8.0
cuda129_generic_h8ef9432_200 (200)
17 days ago
46.31 KB
linux-aarch64
N/A
pytorch-gpu-2.7.1-cuda128_mkl_h2fd0c33_304.conda
2.7.1
cuda128_mkl_h2fd0c33_304 (304)
2 months ago
47.64 KB
win-64
N/A
pytorch-gpu-2.7.1-cuda129_mkl_h43a4b0b_304.conda
2.7.1
cuda129_mkl_h43a4b0b_304 (304)
2 months ago
46.88 KB
linux-64
N/A
pytorch-gpu-2.7.1-cuda129_mkl_h43a4b0b_303.conda
2.7.1
cuda129_mkl_h43a4b0b_303 (303)
3 months ago
47.02 KB
linux-64
N/A
pytorch-gpu-2.7.1-cuda128_mkl_h2fd0c33_303.conda
2.7.1
cuda128_mkl_h2fd0c33_303 (303)
3 months ago
47.75 KB
win-64
N/A
pytorch-gpu-2.7.1-cuda129_mkl_h43a4b0b_302.conda
2.7.1
cuda129_mkl_h43a4b0b_302 (302)
3 months ago
46.89 KB
linux-64
N/A
pytorch-gpu-2.7.1-cuda128_mkl_h2fd0c33_302.conda
2.7.1
cuda128_mkl_h2fd0c33_302 (302)
3 months ago
47.73 KB
win-64
N/A
pytorch-gpu - conda-forge