Instalar Pytorch en Windows

En este artículo, aprenderemos cómo instalar Pytorch en Windows. PyTorch es una biblioteca de aprendizaje automático de código abierto basada en la biblioteca Torch, que se utiliza para aplicaciones como la visión artificial y el procesamiento del lenguaje natural, desarrollada principalmente por el laboratorio de investigación de IA de Facebook. Es un software gratuito y de código abierto publicado bajo la licencia BSD modificada.

Método 1: Usar pip

Paso 1: compruebe si Python ya está instalado ingresando el siguiente comando en el símbolo del sistema.

python –versión

Si este comando se ejecuta con éxito y podemos obtener una versión de Python, entonces podemos continuar o instalar python consultando ¿Cómo instalar Python en Windows?

Paso 2: compruebe si pip ya está instalado ingresando el siguiente comando en el símbolo del sistema.

pip –versión

Si este comando se ejecuta con éxito y podemos obtener una versión de pip, entonces podemos continuar o instalar pip consultando este artículo Descargar e instalar la última versión de pip .

Paso 3: Ingrese cualquiera de los siguientes comandos (según las especificaciones de su sistema) para instalar la última versión estable de Pytorch.

1. Plataforma informática: CUDA 10.2, la versión del controlador Nvidia debe ser >= 441.22

pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

2. Plataforma informática: CUDA 11.1, la versión del controlador Nvidia debe ser >= 456.38

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

3. Plataforma informática: CPU

 pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

Paso 4: compruebe si Pytorch se instaló correctamente ingresando el siguiente comando en el símbolo del sistema.

pip3 mostrar antorcha

Método 2: Usar conda

Paso 1: verifique si conda está instalado ingresando el siguiente comando en Anaconda Prompt.

conda –versión

Si este comando se ejecuta con éxito y podemos obtener una versión conda, entonces podemos continuar o instalar Anaconda consultando este artículo ¿Cómo instalar Anaconda en Windows?

Paso 2: Abra Anaconda Prompt en modo Administrador e ingrese cualquiera de los siguientes comandos (según las especificaciones de su sistema) para instalar la última versión estable de Pytorch.

1. Plataforma informática: CUDA 10.2, la versión del controlador Nvidia debe ser >= 441.22

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

2. Plataforma informática: CUDA 11.1, la versión del controlador Nvidia debe ser >= 456.38

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

3. Plataforma informática: CPU

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Paso 3: compruebe si Pytorch se instaló correctamente ingresando el siguiente comando en el indicador de Anaconda.

lista conda -f pytorch