网站菜单

win10系统极简安装Tensorflow2.1—gpu版本流程

折腾了半天环境,最后发现其实很简单。

官网文档说2.0以后不用单独安装GPU,但是很扯犊子!

特此记录一下以免以后再次踩坑

Anaconda创建一个新的环境:

conda install tensorflow-gpu==2.1.0
#1.13.1

它会自动下载相关依赖包,包括cuda,cudnn。

之后再手动安装各种库:

conda install -c anaconda pandas numpy gensim keras matplotlib nltk scikit-learn pydot smart_open

显示评论 (1)

文章评论

  • Jasuit

    难得还更新了一篇咧

    • 本文作者
    • 回复

相关推荐

爱芯元智AX650N板端验证YoloV5s

笔者的AX650N为AXCL PCIe 算力卡形态;如果你是 SoC 开发板(AXera-Pi/AX650N 等),我在文末也放了 SoC 的做法。 1. 安装运行库 在插卡的主机上(Ubuntu/D…

爱芯元智AX650N部署YoloV5s全流程

1. 训练模型 克隆 yolov5的代码,并安装 yolov5 所需的 python包 以及 onnx,这个就不多赘述了。 与瑞芯微需要特定分支的yolov5(或者修改网络结构)不同,直接使用官方的版…