Pytorch是一个用户友好的python的深度学习框架,类似于tensorflow,Caffe,keras。
目前,Pytorch可支持windows,linux,mac系统。
本人的安装环境是:操作系统 Win10,已经安装过 Anaconda (anaconda的安装可以自行google),pytorch的cpu版本。
打算安装GPU版本的,“需要先安装cuda,再根据cuda相应版本安装相应的pytorch版本”(资料2)。如果电脑是AMD卡,那就不能使用cuda了。
正如参考资料1所说,“安装 Anaconda 完毕后,我们在安装 PyTorch 之前最好先创建一个 pytorch 的虚拟环境。” 这相当于给pytorch单独的一块空间,存放各种包依赖。不会影响到你之前的项目,可以放心大胆安装,即便崩了,也不会对别的项目产生影响。
1. 在anaconda创建一个虚拟环境 (详细操作见参考资料1)
2. 在pytorch官网安装
(官网有命令行,你只需把你电脑参数设置以下就好,直接复制粘贴到anaconda的虚拟环境中)
3. 检查是否安装成功
(网上很多帖子都记录了各种心酸经历和bug,但是目前pytorch好像已经把bug解决了。如图,安装成功)
参考资料:
1. WIn10+Anaconda 环境下安装 PyTorch 避坑指南 - 红色石头的专栏 - CSDN博客
https://blog.csdn.net/red_stone1/article/details/86669362
2. pytorch学习(一)windows10下anaconda+python3.6安装pytorch(cpu版本),在pycharm中使用 - weixin_41783335的博客 - CSDN博客
https://blog.csdn.net/weixin_41783335/article/details/82828334