相信大家都是来找安装教程的,我们直接进入正题。
本文发布时间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,自动会有这个文件出现)
然后就装完了。简单吗。测试一下:
如果没有GPU,那其实安装已经结束了,但我相信大部分人,都是要gpu版本的。
去安装一下cuda8+cudnn6.或者cuda9+cudnn7。我是前者,因为我装过TensorFlow只支持cuda8。网上教程很多,不过坑,其实只有一个。安装cuda请选择自定义安装,只勾选cuda,其他一律不选!