2019年3月10日 星期日

Pytorch 在 Anaconda windows10 CUDA9 GTX1060 GPU 環境下安裝

  1. 先到pytorch 官網 https://pytorch.org/
  2. 點選 Get start https://pytorch.org/get-started/locally/
    我的python 版本是安裝3.6版的,CUDA 安裝是9.0版,接著複製下面那段文字。
    conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

  3. 開始 =>Anaconda3 => 開起 Anaconda Prompt
    激活先前安裝的環境,我自己的環境名稱為(pypy)
    在prompt 中輸入 $ activate pypy 如下:
    激活環境(pypy)後,再輸入$ conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
  4. 都安裝完後,來測試看看是否正確安裝:

    輸入$ python
    再輸入以下編碼:
    >>> from __future__ import print_function
    >>> import torch
    >>> x = torch.rand(5, 3)
    >>> print(x)

    會得到output如下:
    tensor([[0.0598, 0.0296, 0.2705],
            [0.5359, 0.8694, 0.3644],
            [0.8122, 0.4435, 0.0141],
            [0.7665, 0.4349, 0.3779],
            [0.5709, 0.5939, 0.3551]])
    表示已正確安裝。

  5. 接著就能在我們的IDE下寫段編碼測試測試
    import torch
    x = torch.rand(5, 3)
    print(x)

    >>>tensor([[0.3140, 0.0046, 0.6260],
                     [0.1289, 0.4955, 0.4758],
                     [0.1680, 0.4328, 0.9986],
                     [0.5861, 0.3964, 0.8033],
                     [0.9430, 0.3704, 0.5186]])

沒有留言:

張貼留言