PyTorch - 深度學習 AMI
啟動 PyTorch安裝 PyTorch的夜間構建（實驗性）其他教學

本文為英文版的機器翻譯版本，如內容有任何歧義或不一致之處，概以英文版為準。

PyTorch

啟動 PyTorch

當穩定的 Conda 架構套件發行時，會在DLAMI. 如果您想要執行最新、未經測試的每夜組建，您可以手動安裝 PyTorch的夜間構建（實驗性）。

若要啟用目前安裝的架構，請遵循 Conda 深度學習AMI中的這些指示。

對 PyTorch 於使用CUDA和 MKL-的 Python 3DNN，運行以下命令：

$ source activate pytorch_p310

啟動終 iPython 端機。

(pytorch_p310)$ ipython

運行一個快速 PyTorch 程序。

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

您應該會看到列印初始隨機陣列、接著是其大小，然後新增另一個隨機陣列。

安裝 PyTorch的夜間構建（實驗性）

如何 PyTorch 從夜間構建安裝

您可以AMI使用 Conda 將最新 PyTorch 版本安裝到深度學習中的一個或兩個 PyTorch Conda 環境中。

    • （Python 3 的選項）-激活 Python 3 PyTorch 環境：

      $ source activate pytorch_p310

  2. 其餘步驟假設您使用的是 pytorch_p310 環境。刪除當前安裝的 PyTorch：

    (pytorch_p310)$ pip uninstall torch

    • （GPU實例的選項）- PyTorch 使用 CUDA .0 安裝最新的夜間構建：

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

    • (CPU執行個體的選項)-安裝最新的每晚組建的執行個體，不 PyTorch GPUs需要執行下列指令：

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

  4. 若要確認您已成功安裝最新的夜間組建，請啟動IPython終端機並檢查的 PyTorch版本。

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

    輸出應該會列印類似於 1.0.0.dev20180922 的內容

  5. 要驗證 PyTorch 每晚構建 PyTorch是否與示例一起運行良好，您可以從的示MNIST例存儲庫中運行測試腳本：

    (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

其他教學

有關更多教程和示例，請參閱框架的官方PyTorch 文檔，文檔和PyTorch網站。