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

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