步骤:
一. 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
- 完整配置了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命令,一切正常,问题解决。
(注:我台式机上述方式有效,笔记本上无效,且笔记本上pip安装部分包有效如pandas,部分无效如lightgbm或TensorFlow直接报pip版本对不上)
解决方式2(推荐):参考:https://pip.pypa.io/en/stable/installing/
- 使用压缩包解压的方式安装python
- 完整配置了python的相关环境变量
- 若要卸载pip则执行python -m pip uninstall pip
- 执行:curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
- 执行:python get-pip.py
- 添加环境变量path:D:\Program Files (x86)\python-3.7.6-embed-amd64\Scripts\
Windows可能要一两分钟后才会自动刷新path,可以cmd下where pip查看是否成功添加pip;
二. 检查自己显卡版本、驱动(这台电脑-属性-设备管理器-显示适配器)然后解压CUDNN,后面放到CUDA要用到(我这里是放到C:\ProgramData\NVIDIA GPU Computing Toolkit\v11.1 下面include、lib、bin三个文件夹会自动合并,其实也就是CUDNN多几个文件而已),确定自己能用的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 未知
下载链接(链接上写的cuDNN Library for Windows (x86),实际下载是 cudnn-11.2-windows-x64-v8.1.1.33.zip):https://developer.nvidia.com/rdp/cudnn-download
具体参考:https://tensorflow.google.cn/install/source_windows#gpu
三. cuda配置,直接下载安装(环境变量自动好了的,TF1我不用复制CUDNN也能跑,但是TF2要把上步的CUDNN复制过来)
下载网站: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。
附:cudnn 8.1.1 + CUDA11.1 + TensorFlow2.4.0 我后来试了也可以,就是开始pycharm 运行时候有6个DLL加载成功,还有cudnn64_8.dll和cusolver64_10.dll失败,前者在CUDNN里面手动查看文件是有的,后者还没找到。将CUDNN那三个文件夹放到CUDA下,cudnn64_8.dll加载就成功了,还剩个cusolver64_10.dll问题,去网上下载后放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin 就好了。
参考:
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
https://blog.csdn.net/weixin_45284801/article/details/109691589