安裝環境:
系統環境: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