各种安装 TensorFlow 环境踩坑
自己的笔记本安装
今年在美国入手了一个笔记本 Lenovo Legion Y530 (国行好像叫Y7000p) ,就随意尝试一下在自己的 laptop 上安装一下深度学习环境,顺便帮女朋友做个作业。
具体安装方法请参考这位老哥的博客 Win 7/10下Anaconda/Python /TensorFlow-GPU/VS Code安装配置 - senshuran的博客 - 优快云博客,几乎反正安装都没什么好说的,安装过程都是没什么难点的。
我的电脑配置 Windows 10 家庭版 1809,8代 i7 , 双通道 16G 内存,Nvidia GTX 1050 Ti 4GB 显卡。
Python环境准备
我的电脑是 Win10 家庭版,我之前为了做学校的作业,所以已经在电脑上安装过 VSCode 和 Anaconda 3,还在其中创建了 python 2.7 的虚拟环境。
查询版本
事先看了一下好像 Windows 版本的 TensorFlow 都只支持 python 3.5 以上的版本,顺便查询一下对应的 CUDA 版本。
第一个坑
一开始因为 Anaconda 3 默认的是 python 3.7,所以我选择了支持它的 CUDA 10.1 + TensorFlow-gpu 1.13.1,结果装完之后运行的时候疯狂提示缺少什么 DLL 支持。不管使用什么方法都没有用,我也早都装过什么网上说的 VS2015 C++ 那个什么包,各种依赖环境应该都是有的,依然没有用。
最后还是乖乖的降版本,选择了 CUDA 10.1 + TensorFlow-gpu 1.13.1,果然在 import tensorflow 的时候通过了;然后我试了一下 hello world ,又出错了。
第二个坑
选择 CUDA 10.1 + TensorFlow-gpu 1.13.1 的时候报错我现在有点忘了,大致是说什么 h5py 和什么另一个包产生了冲突,网上说什么要重新安装,把 h5py 和那个包都降版本,尝试了一下版本根本没办法匹配,降版本无济于事。
最终还是找到了一个同样使用 1050 Ti 的老哥写的安装博客 win10+1050ti+CUDA9.0+cudnn9.0的tensorflow-gpu环境搭建 - 猪八戒吃胡萝卜 - 优快云博客 ,所以我也选择接着重装,用的离线包,在去看女朋友的路上在火车上装的。最终成功的版本是 CUDA 9.0 + cuDNN 7.6.0 for CUDA 9.0 + TensorFlow-gpu 1.11.0 。当然,因为这个 tf 的版本只支持 python 3.6 ,所以我重新创建了一个支持 python 3.6 的 Anaconda 虚拟环境:
conda create -n tensorflow python=3.6 -y
以上。暂时就这么多。
实习的公司的电脑要保密也不能多说,等哪天想写了再补。。。