windows下安装PyTorch0.3.0

相信大家都是来找安装教程的,我们直接进入正题。

本文发布时间2018/1/6,由于PyTorch更新很快,所以如果距离此时间过长,请慎重参考。

首先先声明:希望大家把python版本设置为3.5.因为在windows上安装深度学习框架,无论是TensorFlow还是Pytorch,都不支持python2,由于TF不支持3.6,所以我推荐大家使用3.5.不过3.6的安装方法也会给大家,大家也可以试试。

windows版本要求64位。32位的同学。。。可以更新了。已经2018年了。32位系统不能存储单个文件大于4G的。。

然后,大家pip应该有吧,这个教程很简单,不需要conda也不需要别的乱七八糟的,就最基础的,pip就行。

然后去网盘下载:

链接:https://pan.baidu.com/s/1dUrI3O 密码:lahq

有两个版本一个对应3.5一个对应3.6.然后下载以后,在这个路径下 pip install "torch-0.3.0b0+591e73e-cp35-cp35m-win_amd64.whl"(输入pip install 然后点击一下或多下tab,自动会有这个文件出现)

然后就装完了。简单吗。测试一下:

windows下安装PyTorch0.3.0_第1张图片

如果没有GPU,那其实安装已经结束了,但我相信大部分人,都是要gpu版本的。

去安装一下cuda8+cudnn6.或者cuda9+cudnn7。我是前者,因为我装过TensorFlow只支持cuda8。网上教程很多,不过坑,其实只有一个。安装cuda请选择自定义安装,只勾选cuda,其他一律不选!然后英伟达驱动安装最新的低一版就可以(不放心最新版),驱动安装可以在cuda前也可以在cuda后。至于cudnn,复制黏贴,很简单。下载不下来的话,可以联系我。

如果你在安装pytorch之前没有装cuda,那么你需要先

pip uninstall torch
然后再重新装一遍,测试的话:

windows下安装PyTorch0.3.0_第2张图片

然后,你们还可以试试具体案例(官方demo有的跑不起来,需要略作修改,可能是因为PyTorch版本更新太快的原因),比如我写的一个训练好的ResNet:我的github,直接下载到一个文件夹下,

python load_resnet.py
然后它会下载训练集,给出结果。

遇到任何问题,欢迎评论或私信我。

你可能感兴趣的:(安装类教程)