首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

Windows下安装PyTorch0.4.0

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.运行错误

  • 发表于:
  • 原文链接http://kuaibao.qq.com/s/20180512G0Y4QX00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券