win10+cpu 安裝 pytorch

安裝環境:
系統環境:win10 CPU
python環境:Python3.6.5
使用pip進行安裝:

pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl
pip install torchvision

測試是否安裝成功:

import torch
import numpy as np
np_data = np.arange(6).reshape((2,3))
torch_data = torch.from_numpy(np_data)
print('np_data',np_data)

當結果爲np_data [[0 1 2]
[3 4 5]]
時,即爲安裝成功。

參考博客:https://blog.csdn.net/u014439564/article/details/80270996
https://blog.csdn.net/u014439564/article/details/79517642

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章