- 先到pytorch 官網 https://pytorch.org/
- 點選 Get start https://pytorch.org/get-started/locally/
我的python 版本是安裝3.6版的,CUDA 安裝是9.0版,接著複製下面那段文字。
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
- 開始 =>Anaconda3 => 開起 Anaconda Prompt
激活先前安裝的環境,我自己的環境名稱為(pypy)
在prompt 中輸入 $ activate pypy 如下:
激活環境(pypy)後,再輸入$ conda install pytorch torchvision cudatoolkit=9.0 -c pytorch - 都安裝完後,來測試看看是否正確安裝:
輸入$ 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]])
表示已正確安裝。
- 接著就能在我們的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]])
2019年3月10日 星期日
Pytorch 在 Anaconda windows10 CUDA9 GTX1060 GPU 環境下安裝
訂閱:
張貼留言 (Atom)
沒有留言:
張貼留言