安装环境:
系统环境: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