美文网首页
【机器学习笔记】——TensorFlow的安装(cuda10.0

【机器学习笔记】——TensorFlow的安装(cuda10.0

作者: 阿脸阿脸 | 来源:发表于2018-12-11 19:28 被阅读0次

    我是win7,显卡GTX1050Ti,选的cuda10.0,cuDNN7.4

    tensorflow分为CPU版本和GPU版本,CPU版本运行较慢。

    1 安装anaconda3(python3.6)

    2 安装tensorflow

    2.1 CPU版本安装

    不依赖于其他,直接在cmd中 conda install tensflow

    2.2 GPU版本安装

    1)是否支持GPU版本

    在控制面板-设备管理-显示配置器中查看是否有支持GPU的显卡

    2)安装cuda

    cuda最新(10.0)版本:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=7&target_type=exelocal

    选择适合电脑系统win7的cuda安装包:

    如果需要其他版本的cuda:https://developer.nvidia.com/cuda-toolkit-archive

    安装步骤一步一步来就行。

    3)安装cuDNN

    cuDNN版本需要与cuda匹配

    https://developer.nvidia.com/rdp/cudnn-download

    如果需要其他版本的cuDNN:https://developer.nvidia.com/rdp/cudnn-archive?tdsourcetag=s_pctim_aiomsg

    解压后,是这些:

    将这三个复制至cuda的安装目录下就可以了。

    4)安装tensorflow

    conda install tensflow-gpu

    相关文章

      网友评论

          本文标题:【机器学习笔记】——TensorFlow的安装(cuda10.0

          本文链接:https://www.haomeiwen.com/subject/nbgehqtx.html