步骤:
一. Python环境搭建:
Python官网下载Python3.7.2压缩包ZIP(注意我们一般用embeddable免安装压缩包)解压安装Python3;再安装pip:
curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
python get-pip.py
然后进到pycharm中使用刚下载的Python3无法pip添加包,但是Python2一切正常,报错No module named pip(直接install packpackaging tools或者python get-pip.py
报 No module named 'setuptools'问题同理,需要对应版本的pip);
解决方式1:参考: https://segmentfault.com/q/1010000016236939
python get-pip.py
安装pip经过观察,运行python get-pip.py
后python的安装目录(对于题主来说是"D:\z_tools\python-3.7.0"
)多了文件夹Lib\site-packages
,尝试将其中的pip目录复制到python的安装目录,再次运行pip命令,报错消失。
撤销复制,修改python安装目录下的python3x._pth
文件,新添加一行Lib\site-packages
。
再次运行pip命令,一切正常,问题解决。
(注:我台式机上述方式有效,笔记本上无效,且笔记本上pip安装部分包有效如pandas,部分无效如lightgbm或TensorFlow直接报pip版本对不上)
解决方式2(推荐):参考:https://pip.pypa.io/en/stable/installing/
Windows可能要一两分钟后才会自动刷新path,可以cmd下where pip查看是否成功添加pip;
二. 检查自己显卡版本、驱动(这台电脑-属性-设备管理器-显示适配器),确定自己能用的tensorflow-gpu版本,版本对应关系是:
tensorflow-gpu v1.9.0 | cuda9.0 | cuDNN7.1.4可行 | 备注:7.0.4/ 7.0.5/ 7.1.2不明确
tensorflow-gpu v1.8.0 | cuda9.0 | cuDNN 不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4
tensorflow-gpu v1.7.0 | cuda9.0 | cuDNN 不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4
tensorflow-gpu v1.6.0 | cuda9.0 | cuDNN 不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4
tensorflow-gpu v1.5.0 | cuda9.0 | cuDNN 不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4
tensorflow-gpu v1.4.0 | cuda8.0 | cuDNN 6.0 | 备注:6.0正常使用, 7.0.5不能用,5.1未知
tensorflow-gpu v1.3.0 | cuda8.0 | cuDNN 6.0 | 备注:6.0正常使用, 7.0.5不能用,5.1未知
tensorflow-gpu v1.2.0 | cuda8.0 | cuDNN 5.1 | 备注:5.1正常使用, 6.0/ 7.0.5 未知
tensorflow-gpu v1.1.0 | cuda8.0 | cuDNN 5.1 | 备注:5.1正常使用, 6.0/ 7.0.5 未知
三. cuda配置,直接下载安装
下载网站:https://developer.nvidia.com/cuda-90-download-archive
三.. 安装Tensorflow-gpu
pip unistall tensorflow-gpu
过程有点问题:装到一半后发现装不了termcolor包,然后绕道了干脆用anaconda管理环境好了,到anaconda官网下载最新的版本,然后发现anaconda是能装那个包的。。。具体原因不深究。但是anaconda下安装tensorflow-gpu要注意手动选择版本,不然自动最新的版本会报错的:
这个问题我直接卸载自动选择的版本,重装手动选择版本tensorflow-gpu v1.9.0 | cuda9.0 | cuDNN7.1.4(注我的是英伟达Q400显卡,2019.1.1最新版驱动),然后就好了:
如果还是喜欢用pycharm,则将anaconda3\python.exe导入到pycharm - setting里面引用就好了,如果是老版本anaconda3则是anaconda3\bin\python.exe。
参考:
https://blog.csdn.net/lifuxian1994/article/details/81103530
https://www.cnblogs.com/fanfzj/p/8521728.html
https://pypi.org/project/tensorflow-gpu/#files
https://developer.nvidia.com/rdp/cudnn-download