2018年4月25号,PyTorch 官方发布 0.4.0 版本,该版本的 PyTorch 有多项重大更新,其中最重要的改进是官方支持 Windows (详细改动见Pytorch官方GitHub):Windows各种安装方式汇总:
Windows用户能直接通过conda、pip和源码编译三种方式来安装Pytorch了,假如我们用pip来安装的话:
安装GPU版本,如:Windows+pip+python 3.6+CUDA 8 ,只需在cmd中输入:
如果电脑没有英伟达显卡,那就只能安装CPU版本的,如:Windows+pip+python 3.6 ,在cmd中输入:
最后需要安装 torchvision,我是通过本地安装的方式进行安装的,下载地址:(https://pypi.python.org/pypi/torchvision/0.2.1)下载好之后,用cmd进入torchvision所在的文件夹下,输入下面的命令安装:
测试一下是否安装好了,看看能不能用GPU加速:
安装后会出现一些问题,官方文档针对一些问题给出了解决方案,在知乎上有大神也给出了一些解决方案。1.import torch失败
2.运行错误
领取专属 10元无门槛券
私享最新 技术干货