1. conda
En primer lugar, necesitamos tener conda, el proceso de instalación y configuración de miniconda se puede ver en este http://t.csdn.cn/ShcNb
2. Ingresa al sitio web oficial
Seleccione la versión correspondiente y copie el siguiente comando
conda install pytorch torchvision torchaudio -c pytorch
3. Comience a crear un entorno virtual y descargue pytorch
en la terminal, ingrese
conda create -n torch-gpu python=3.9
conda activate torch-gpu
conda install pytorch torchvision torchaudio -c pytorch
El primer comando es crear un entorno virtual llamado torch-gpu, python versión 3.9
El segundo comando es para activar el entorno.
El tercer comando es instalar pytorch
4. prueba
python
import torch
torch.__version__
torch.device('mps')