Tensorflow安装过程

Published on with 0 views and 0 comments

随着硬(you)件(xi)更新换代越来越快,在这记录安装 tensorflowgpu 支持的过程概要

安装过程

  1. 确定你电脑的 GPU 型号,只支持绿卡。

  2. 安装 anaconda(可选)

  3. pip 安装 tensoflow-gpu,这一步关系到下一步的版本号

  4. 确定 GPU 型号所适配的 CUDA 版本,与 cuDNN 版本
    这里讲解一下,CUDA 相当于显卡的并行计算专用驱动,cuDNN 相当于程序需要的 API
    根据 tensorflow 包版本确定需要的 CUDA 版本和 cuDNN 版本
    下载指定版本的 CUDA 版本包
    先注册登录,然后下载指定版本的 cuDNN

  5. 开始安装 CUDA 和 cuDNN
    安装 CUDA 时建议选自定义,驱动必选,因为它所需的驱动版本比你一般玩游戏需要的都要新。然后将 VS 和它的一个依赖去掉,不然会又大又慢
    如果安装好了,可以在 cmd 尝试输入

    1nvcc -V
    

    cuDNN 可以不叫安装,解压后可以直接覆盖到 CUDA 的安装目录下(bin,include,lib/x64),下一步的环境变量就比较好配置

  6. 配置系统环境变量 Path
    可能是直接配好的,也可能需要你添加一下,具体路径仅供参考(CUDA 和 cuDNN 放在一起的情况下)
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64

完结撒花


标题:Tensorflow安装过程
作者:Dadong-Zhang
地址:https://keithzhang.cn/articles/2020/12/08/1607433792513.html