grapycal_torch is an interface for interacting with PyTorch. With it, Grapycal is able to manipulate PyTorch Tensors, train neural networks, etc….

Installation

grapycal ext install torch

The installation may take a while. If your GPU supports CUDA (nvidia GPUs), the installation command may install PyTorch with CUDA support, which is usually larger than 2GB.

After successfully installing the extension, you’ll be able to import it.