美文网首页
Windows 10 下安装tensorflow gpu 1.1

Windows 10 下安装tensorflow gpu 1.1

作者: shiguang116 | 来源:发表于2019-11-13 20:35 被阅读0次

    Windows 10 下安装tensorflow比较简单,只要解决好兼容性问题即可。

    1. 首先安装cuda 10.0
      下载地址cuda 10.0 windows 10,下载完毕直接安装即可

    cuda 10.0中已经集成了兼容的显卡驱动,因此如果是新装的windows 10 系统,不需要提前安装显卡驱动。

    1. 安装cudnn
      下载地址
      https://developer.nvidia.com/rdp/form/cudnn-download-survey

    下载cudnn需要先注册一下,登陆后才可以进入下载界面,然后选择对应的版本下载。

    下载下来的文件为cudnn-10.0-windows10-x64-v7.6.5.32.zip,将压缩包里的文件解压缩后放到cuda 10.0的安装目录下 (一般为
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\)
    覆盖同名的目录即可。

    1. 安装Anaconda
      这里选择包含python 3.6的最终版本的Anaconda
      Anaconda3-5.2.0-Windows-x86_64.exe

    注意安装过程 选中将anaconda添加到系统的环境变量 的选项。

    1. conda 添加清华的源(为了加快后续的安装速度,可选)
      打开命令行执行 conda config --set show_channel_urls yes 会在 用户目录(C:\User\用户名)下生成 .condarc文件。
      用记事本打开.condarc文件,输入
    channels:
      - defaults
    show_channel_urls: true
    default_channels:
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
      - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
    custom_channels:
      conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
      simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
    

    保存文件即可

    参考https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

    1. 安装tensorflow
      打开命令行输入
      conda install tensorflow-gpu=1.14
      安装完毕

    如果需要安装keras,继续输入命令
    conda install keras=2.15
    即可

    相关文章

      网友评论

          本文标题:Windows 10 下安装tensorflow gpu 1.1

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