PyTorch

Ativando PyTorch

Quando um pacote Conda estável de uma estrutura é lançado, ele é testado e pré-instalado no. DLAMI Se desejar executar as mais recentes compilações noturnas não testadas, você pode PyTorchInstall's Nightly Build (experimental) manualmente.

Para ativar a estrutura atualmente instalada, siga estas instruções em seu Deep Learning AMI com Conda.

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

$ source activate pytorch_p310

Inicie o iPython terminal.

(pytorch_p310)$ ipython

Execute um PyTorch programa rápido.

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.

PyTorchInstall's Nightly Build (experimental)

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

Você pode instalar a PyTorch versão mais recente em um ou em ambos os ambientes PyTorch Conda em seu Deep Learning AMI com o Conda.

1. • (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 atualmente instalado PyTorch:

   (pytorch_p310)$ pip uninstall torch

3. • (Opção para GPU instâncias) - Instale a versão noturna mais recente do PyTorch with CUDA .0:

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

   • (Opção para CPU instâncias) - Instale a versão noturna mais recente de PyTorch for 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 última compilação noturna, inicie o IPython terminal 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 se a compilação PyTorch noturna funciona bem com o MNIST exemplo, você pode executar um script de teste no repositório de exemplos PyTorch da:

   (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 mais tutoriais e exemplos, consulte os documentos oficiais, a PyTorch documentação e o site da estrutura. PyTorch