Win10下的pytorch,torch简明安装(使用conda命令)

Pytorch简介与Win10安装

Pytorch是facebook开发的深度学习库,其目标是想成为深度学习领域整合GPU加速的numpy。话说开发还是要用Linux环境啊,无奈自己笔记本装不上Ubuntu,如今pytoch目前还没有对Windows的官方支持(计划0.4版本开始支持,目前是0.3.1),所幸的是知乎用户蒲嘉宸一直在提供适合windows的包。


首先,我们根据自己的环境下载对应包:
https://pan.baidu.com/s/1dF6ayLr#list/path=%2Fpytorch

1.注意这里因为py3版本只支持3.5.3与3.6.2,如果电脑python版本不一致,使用Anaconda新建一个名为py36的新环境,打开命令提示符,输入以下命令:

conda create -n py36 python=3.6.2/3.5.3 numpy pyyaml mkl

2.一般在网盘中下载的版本为:

pytorch-0.3.1-py36_cuda80_cudnn6he774522_2.tar.bz2

当然,要根据自己的windows版本和环境选择下载,这个版本对应的是win10的cuda 8.0与cudnn 6环境,如果不知道,可以试一下这个

3.下载成功后,首先激活py36,然后执行pytorch的安装(以管理员身份运行cmd)

activate py36
conda install (路径)\pytorch-0.3.1-py36_cuda80_cudnn6he774522_2.tar.bz2

4.不出意外即可安装成功,继续安装torchvision,先下载torchvison

https://pypi.python.org/pypi/torchvision
笔者选择的是目前最新的torchvision-0.2.0-py2.py3-none-any.whl

5.安装tochvision

pip install torchvision-0.2.0-py2.py3-none-any.whl

6.测试,先将环境切换到py36

activate py36
python
>>> import torch
>>> import torchvision
>>> a=torch.rand(1,2)
>>> b=a.cuda()
>>> print(a,b)
 0.5278  0.2514
[torch.FloatTensor of size 1x2]
 0.5278  0.2514
[torch.cuda.FloatTensor of size 1x2 (GPU 0)]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值