Aller au contenu

torch

Installation

  • Créer un environnement "torch"
conda create --name torch 
conda activate torch
  • Installer pytorch avec CUDA

pytorch.org - Get Started - Start Locally

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia