PyTorch

Mengaktifkan PyTorch

Ketika paket Conda yang stabil dari kerangka kerja dirilis, itu diuji dan diinstal sebelumnya pada file. DLAMI Jika Anda ingin menjalankan build malam terbaru yang belum teruji, Anda dapat secara manual. Instal PyTorch Nightly Build (eksperimental)

Untuk mengaktifkan kerangka kerja yang saat ini diinstal, ikuti petunjuk ini di Deep Learning Anda AMI dengan Conda.

Untuk PyTorch pada Python 3 dengan CUDA dan MKL -DNN, jalankan perintah ini:

$ source activate pytorch_p310

Mulai iPython terminal.

(pytorch_p310)$ ipython

Jalankan PyTorch program cepat.

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

Anda akan melihat array acak awal dicetak, kemudian ukurannya, dan kemudian penambahan array acak lainnya.

Instal PyTorch Nightly Build (eksperimental)

Cara menginstal PyTorch dari build malam

Anda dapat menginstal PyTorch build terbaru ke salah satu atau kedua lingkungan PyTorch Conda di Deep Learning Anda AMI dengan Conda.

1. • (Opsi untuk Python 3) - Aktifkan lingkungan Python 3: PyTorch

     $ source activate pytorch_p310

2. Langkah-langkah yang tersisa menganggap Anda menggunakan pytorch_p310 lingkungan. Hapus yang saat ini diinstal PyTorch:

   (pytorch_p310)$ pip uninstall torch

3. • (Opsi untuk GPU instance) - Instal build malam terbaru dengan.0: PyTorch CUDA

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

   • (Opsi untuk CPU instance) - Instal build malam terbaru PyTorch untuk instance tanpa: GPUs

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

4. Untuk memverifikasi bahwa Anda telah berhasil menginstal build nightly terbaru, mulai IPython terminal dan periksa versi. PyTorch

   (pytorch_p310)$ ipython

   import torch
   print (torch.__version__)

   Outputnya harus mencetak sesuatu yang mirip dengan 1.0.0.dev20180922

5. Untuk memverifikasi bahwa PyTorch nightly build berfungsi dengan baik dengan MNIST contoh, Anda dapat menjalankan skrip pengujian dari repositori contoh 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

Lebih Banyak Tutorial

Untuk tutorial dan contoh lebih lanjut, lihat dokumen resmi kerangka kerja, PyTorch dokumentasi, dan PyTorchsitus web.