FICHA · AUR

python-torchvision-rocm

Datasets, transforms, and models specific to computer vision (with ROCM support)

  • TorchVision ROCm library
  • LIBRARY
  • Dependency only
official+codex · reviewed · Jun 3, 2026 description in en

Description

Computer-vision datasets, transforms, and pretrained model utilities can be used from PyTorch with ROCm support. Researchers and ML engineers can train or evaluate vision pipelines on AMD GPU stacks. It processes image data that may contain private scenes or people.

Permissions

Permissions not analysed for this source yet.