关于tensorflow和pytorch安装

环境:windows  7,    python 3.7       Pycharm 2018.2 professional

因为是全部安装完后整理的,所以只是以解决问题为主,很多东西都没有深入去了解原理,仅仅记录下踩坑的过程。

1.把虚拟环境的pip升级,之前不升级,一直报要求检测pip的版本,所以默认升级到最新

我的pycharm里面默认是pip  10.0.1

找到虚拟环境的路径,找到/Scripts下的activate文件,直接拖到cmd窗口里面,即进入了对应的venv,

然后执行 python -m pip install --upgrade pip,   此时会报错,不用管,找到/Lib/site-packages下的pip-version文件夹删除(新的加上旧的两个文件夹,如pip-10.0.1-py3.7.egg,注意pip文件夹不要删),然后继续执行上面的语句,就会安装到最新版本

2.安装TensorFlow

开始是直接使用pycharm里面搜索安装的,安装完成后可以运行,但是运行时会提示一个AVX2警告,个人认为可以无视,但是还是搜了下,在这个网址下载你需要的版本到本地,根据你的python版本,cpu或是gpu,然后根据警告选择了AVX2

https://github.com/fo40225/tensorflow-windows-wheel/commit/c44a045d57341222e428a57f349a34e3b6716aea

然后pip install就可以了,再运行警告消除。

因为我是安装的CPU版本的,所以简单点,如果装GPU版本的,好像要先安装好CUDA和cuDNN(N卡的一个什么东东,没深入研究),安装过程中要注意你的显卡支持的版本,我没有尝试,网上有很多文章可以参考,虽然GPU的矩阵运算能力比CPU要强,但是大神好像都建议菜鸟先玩转了CPU再说,具体里面什么区别也未进一步研究。

3.安装pytorch

刚开始在pycharm里面装,无法安装,各种解决不了,按照网上说的添加数据源,改配置都不行,最后上官网https://pytorch.org,发现其实安装很简单

只要勾选,run command都生成了,但是不知道版本问题还是环境问题,我用pip死活执行不成功,报错指向numpy安装版本过低,但是我已经是用了最新版本了。。。也没有查到具体版本间的对照信息,只能试试先安装anaconda,下载地址,https://www.anaconda.com/,然后配置pycharm指向anaconda的解释器(anaconda干嘛用的没任何概念),执行给出的命令,安装成功

感觉菜鸟学习的时候,网上很多资料可以参考真是好事,谢谢各位前辈的总结,但是同时自己也要学会鉴别和变通

猜你喜欢

转载自blog.csdn.net/kingkingzhu/article/details/88975462