安装的基本配置如下:
windows 8.1 + Anaconda3 + Python3.6 + CPU
默认已安装好 Anaconda3,并创建了 python3.6 的环境
如果不清楚Anaconda的安装及使用,可先参考https://www.jianshu.com/p/eaee1fadc1e9
按照上述网址给出的方法,安装并创建好python3.6的环境
一、官网上给出的安装方法
① 安装 PyTorch 的语句
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
a_np = np.arange(9).reshape((3,3))
print(a)
b_torch = torch.from_numpy(a_np)
print(b)
如果一切顺利,那么测试语句将正常运行,PyTorch安装完毕
由于本人电脑同时装了python2.7和python3.6所以在打开电脑CMD后输入python 显示的是python2.7,因为python2.7中的python.exe文件相比python3.6来说靠前点。
如果要显示Anaconda3中安装的python版本,可以先切换到Anaconda中的python.exe文件夹路径下,再输入python就可以显示Anaconda中py安装的版本号了
二、安装过程中可能出现的问题
① pip 安装时报错: torch-0.4.0-cp36-cp36m-win_amd64.whl is not a supported wheel on this platform
pytorch 安装时需要64位的环境,如果 Anaconda 配置 python 时将其配置成了32位的环境,那么就会报上述的错误
解决方法: 安装Anaconda和配置python环境时,将其配置为64位的环境
② No module named 'torch._C'
解决方法: 将 Anaconda3\envs\**你的python环境**\Lib\site-packages\torch目录下的
_C.cp36-win_amd64.pyd 文件重命名为 _C.pyd
② ImportError: DLL load failed: 找不到指定的模块解决办法
解决方法: 可参考 https://blog.csdn.net/taxuezcy/article/details/80741315
安装GPU版本如下https://blog.csdn.net/qq_40388909/article/details/80504271
安装好以后测试回归的代码:https://blog.csdn.net/zyb228/article/details/80113667
GPU上跑代码https://blog.csdn.net/qq_28444159/article/details/78781201