PyTorch - AWS Deep Learning AMIs
Activación de PyTorchInstalación de una compilación nocturna de PyTorch (experimental)Más tutoriales

PyTorch

Activación de PyTorch

Cuando se lanza un paquete Conda estable de un marco de trabajo, se prueba y se preinstala en la DLAMI. Si desea ejecutar la última compilación nocturna sin probar, puede Instalación de una compilación nocturna de PyTorch (experimental) manualmente.

Para activar el marco de trabajo instalado actualmente, siga estas instrucciones en la AMI de aprendizaje profundo con Conda.

Para PyTorch en Python 3 con CUDA y MKL-DNN, ejecute este comando:

$ source activate pytorch_p310

Inicie el terminal de iPython.

(pytorch_p310)$ ipython

Ejecute un programa rápido en PyTorch.

import torch
x = torch.rand(5, 3)
print(x)
print(x.size())
y = torch.rand(5, 3)
print(torch.add(x, y))

Debería ver la matriz aleatoria inicial, a continuación su tamaño y, por último, la adición de otra matriz aleatoria.

Instalación de una compilación nocturna de PyTorch (experimental)

Cómo instalar PyTorch desde una compilación nocturna

Puede instalar la última compilación de PyTorch en uno o en ambos entornos Conda de PyTorch en su AMI de aprendizaje profundo con Conda.

    • (Opción para Python 3) - Active el entorno PyTorch de Python 3:

      $ source activate pytorch_p310

  2. En los pasos restantes se da por hecho que está utilizando el entorno pytorch_p310. Elimine el entorno de PyTorch instalado actualmente:

    (pytorch_p310)$ pip uninstall torch

    • (Opción para las instancias de GPU) - Instale la última compilación nocturna de PyTorch con CUDA.0:

      (pytorch_p310)$ pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html

    • (Opción para las instancias de CPU) - Instale la última compilación nocturna de PyTorch para las instancias que no tienen ninguna GPU:

      (pytorch_p310)$ pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html

  4. Para verificar que ha instalado correctamente la última compilación nocturna, inicie el terminal de IPython y compruebe la versión de PyTorch.

    (pytorch_p310)$ ipython
    import torch
print (torch.__version__)

    El resultado debería ser similar a 1.0.0.dev20180922

  5. Para verificar que la compilación nocturna de PyTorch funciona bien con el ejemplo de MNIST, puede ejecutar un script de prueba del repositorio de ejemplos de PyTorch:

    (pytorch_p310)$ cd ~
(pytorch_p310)$ git clone https://github.com/pytorch/examples.git pytorch_examples
(pytorch_p310)$ cd pytorch_examples/mnist
(pytorch_p310)$ python main.py || exit 1

Más tutoriales

Si desea ver más tutoriales y ejemplos, consulte la documentación oficial del marco de trabajo, la documentación de PyTorch y el sitio web de PyTorch.