PyTorch

Mengaktifkan PyTorch

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

Untuk mengaktifkan kerangka kerja yang saat ini diinstal, ikuti petunjuk ini pada AMI Pembelajaran Mendalam Anda dengan Conda.

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

$ source activate pytorch_p310

Mulai terminal IPython.

(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 harus melihat array acak awal dicetak, lalu ukurannya, dan kemudian penambahan array acak lainnya.

Inst PyTorch al Nightly Build (eksperimental)

Cara menginstal PyTorch dari build malam

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

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

     $ source activate pytorch_p310

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

   (pytorch_p310)$ pip uninstall torch

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

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

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

     (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 terminal IPython dan periksa versinya. PyTorch

   (pytorch_p310)$ ipython

   import torch
   print (torch.__version__)

   Outputnya harus mencetak sesuatu yang mirip dengan 1.0.0.dev20180922

5. Untuk memverifikasi bahwa build PyTorch nightly berfungsi dengan baik dengan contoh MNIST, Anda dapat menjalankan skrip pengujian dari repositori PyTorch contoh:

   (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

Tutorial Lainnya

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