PyTorch - AMIs de deep learning da AWS
Como ativar o PyTorchInstale a compilação noturna do PyTorch (experimental)Mais tutoriais

PyTorch

Como ativar o PyTorch

Quando um pacote Conda estável de uma estrutura é lançada, ele é testado e pré-instalado na DLAMI. Se desejar executar as mais recentes compilações noturnas não testadas, você pode Instale a compilação noturna do PyTorch (experimental) manualmente.

Para ativar a estrutura instalada no momento, siga estas instruções sobre a AMI de deep learning com Conda.

Para PyTorch em Python 3 com CUDA e MKL-DNN, execute este comando:

$ source activate pytorch_p310

Inicie o terminal iPython.

(pytorch_p310)$ ipython

Execute um programa rápido do PyTorch.

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

Você deve ver a matriz aleatória inicial impressa, seu tamanho e a adição de outra matriz aleatória.

Instale a compilação noturna do PyTorch (experimental)

Como instalar o PyTorch a partir de uma compilação noturna

Você pode instalar a versão mais recente do PyTorch construída em um ou ambos os ambientes PyTorch Conda na AMI de deep learning com o Conda.

    • (Opção para Python 3) - ative o ambiente Python 3 PyTorch:

      $ source activate pytorch_p310

  2. As etapas restantes assumem que você está usando o ambiente do pytorch_p310. Remova o PyTorch atualmente instalado:

    (pytorch_p310)$ pip uninstall torch

    • (Opção para instâncias de GPU) Instale a build noturna mais recente do PyTorch com CUDA.0:

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

    • (Opção para instâncias CPU) - Instale a versão mais recente da compilação noturna do PyTorch para instâncias sem GPUs:

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

  4. Para verificar se você instalou com sucesso a versão mais recente da compilação noturna, inicie o terminal IPython e verifique a versão do PyTorch.

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

    A saída deve imprimir algo semelhante a 1.0.0.dev20180922

  5. Para verificar que a compilação noturna do PyTorch funcione bem com o MNIST de exemplo, você pode executar um script de teste do repositório de exemplos do 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

Mais tutoriais

Para obter mais tutoriais e exemplos, consulte os documentos oficiais da estrutura, Documentação do PyTorch e o site do PyTorch.