Win10安装配置tensorflow
目的
- 只是想在win10安装配置好tensorflow,运行pretrained models 。
- 不训练,训练还是交给linux工作站。
- 为什么要在win10下搞呢?
- 听说win10配置安装很容易;
- 网上教程很多;
- 经常携带的是win10,不妨一试。
安装anaconda
python版本控制,安装package很方便。
最新版带的是python3.7.1, 没必要纠结这个版本问题。因为可以自由创建其它版本的python环境。
安装cuda和cudnn(出错,后面没用到。可能需要自己编译)
cuda和cudnn 的windows版本只能官网下载。
- 提前确认要下载的cuda版本,按照教程发现右击桌面,选nvidia experience找帮助→系统信息→组件→NVIDIA-cuda,查看后面版本发现是10.0。应该是向下兼容。
本来试过cuda10.0,发现arcface代码并不支持,所以选cuda9.0 - 注册登录,下载对应cudnn
发现并不像教程说的那样,把cudnn放在c:\ProgramData目录下然后并把路径添加环境变量path。
完全没必要,直接解压放在cuda9.0的安装目录,因为默认安装cuda时会把bin加到环境变量,所以无需再设置了。
安装tensorflow
去github查python版本,查cuda版本,找对应tensorflow版本。一堆错误。
不如直接用命令安装,网速慢,就指定源
Tensorflow 1.X 安装:
pip3 install tensorflow-gpu==1.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install --index-url https://pypi.douban.com/simple tensorflow
或
pip install --index-url http://mirrors.aliyun.com/pypi/simple/ tensorflow
Tensorflow 2.X 安装: (豆瓣源, 注意要是 https, 有个s )
pip install -i https://pypi.doubanio.com/simple/ tensorflow
最后发现没有编译过cuda,cudnn,还是得装cpu版
报错,难以解决,于是奋起反抗
怒装tensorflow cpu版本
把上面-gpu去掉,装上对应版本,顺利跑起来。