win10安装tensorflow-gp

发布时间:2019-05-24 21:54:08编辑:auto阅读(2148)

    一,本机配置

    Win10 64bit

    NVIDIA GeForce GTX 960M

    Python3.7(Anaconda)

    二,安装CUDA

    亲测,TensorFlow-gpu1.13.1支持cuda10.0的版本,所以我们可直接选择cuda10.0的版本

    Window10下载CUDA10

    安装步骤为:

    1)运行.exe可执行和层序。安装软件会先运行一个系统检查,如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况,系统检查则会报错,无法进行下一步。如果报错了,就最先考虑下是不是自己的显卡不被该版本支持。在安装过程中,它会先自动安装和你电脑对应的NVIDIA驱动。
    2)选择默认安装就行。

    三,安装cuDNN

    cuDNN7.3.1下载

    下载cuDNN需要自己先在以上网址(cuDNN官网)注册个账号,注册成功后登陆进去选择cuDNN7.3.1下载即可。

    下载完成后是一个.zip包,解压后进入目录,会发现三个子目录/bin,/include,/lib/x64。进去后会发现有三个文件,将这三个文件分别拷贝到cuda的安装目录的对应子目录下

    默认cuda的安装目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

    四,配置系统Path环境变量

    检查你的系统Path环境变量中是否有这四项,没有请手动添加

    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
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

    五,安装tensorflow-gpu

    打开Anaconda Prompt,输入

    pip install tensorflow-gpu

    会自动下载并安装tensorflow-gpu

    六,安装测试

    打开pycharm,创建一个test.py文件,输入

    import tensorflow as tf
    
    tf.enable_eager_execution()
    print(tf.__version__)
    hello = tf.constant("hello tensorflow")
    print(hello.numpy())

    出现下列结果表明安装成功

关键字