美文网首页
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