Tensorflow在win10下安装试水

本文记录了在Windows 10系统中安装Tensorflow的过程,包括使用Anaconda进行Python版本控制,尝试安装CUDA和cuDNN但遇到问题,最终选择安装Tensorflow的CPU版本以运行预训练模型。虽然CUDA安装出错,但通过指定Python和CUDA版本,成功安装了Tensorflow的CPU版。
摘要由CSDN通过智能技术生成

目的

  • 只是想在win10安装配置好tensorflow,运行pretrained models
  • 不训练,训练还是交给linux工作站。
  • 为什么要在win10下搞呢?
  1. 听说win10配置安装很容易;
  2. 网上教程很多;
  3. 经常携带的是win10,不妨一试。

安装anaconda

python版本控制,安装package很方便。

官网下载太慢,可以选择清华源

最新版带的是python3.7.1, 没必要纠结这个版本问题。因为可以自由创建其它版本的python环境。

安装cuda和cudnn(出错,后面没用到。可能需要自己编译)

cuda和cudnn 的windows版本只能官网下载。

  1. 提前确认要下载的cuda版本,按照教程发现右击桌面,选nvidia experience找帮助系统信息组件NVIDIA-cuda,查看后面版本发现是10.0。应该是向下兼容。
    本来试过cuda10.0,发现arcface代码并不支持,所以选cuda9.0
  2. 注册登录,下载对应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去掉,装上对应版本,顺利跑起来。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值