步骤:
一. Python环境搭建:
Python官网下载Python3.7.2压缩包ZIP解压安装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;
解决方式:参考: https://segmentfault.com/q/1010000016236939
- 使用压缩包解压的方式安装python
- 完整配置了python的相关环境变量
- 运行
python get-pip.py安装pip - 使用pip得到报错" No module named '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命令,一切正常,问题解决。
二. 检查自己显卡版本、驱动(这台电脑-属性-设备管理器-显示适配器),确定自己能用的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