网站菜单

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

    难得还更新了一篇咧

    • 本文作者
    • 回复

相关推荐

使用Ollama部署本地Deepseek大模型知识库

虽然这篇教程是以 Windows 系统为基础,但其实大多数步骤在其他系统上也可以操作,原理基本一样。 如果你是 Linux 或 Mac 用户,可以稍微根据自己的系统做点调整。 第一步:安装 Ollam…

Yolov5_Seg输出解析

通过矩阵乘法(在代码中称为“matmul”)来计算分割掩码的原因,主要与实例分割网络(例如 YOLOv5 Segmentation)的实现方式有关。这种方法实际上是一种高效的特征图与目标分割系数组合的…