美文网首页
Tensorflow-GPU 环境安装与配置(Windows10

Tensorflow-GPU 环境安装与配置(Windows10

作者: 单名一个冲 | 来源:发表于2019-06-03 18:50 被阅读0次
    一、 N卡安装与配置环境变量
    1. 如果还没有安装 Nvidia CUDA 和 cuDNN ,请先点击查看

    2. Tensorflow-gpu 硬件要求:

    https://tensorflow.google.cn/install/gpu#hardware_requirements

    1. 小编成功配置的版本及环境:

    操作系统:Windows10
    系统位数:64 bit
    GPU:GeForce GTX 1060 6GB
    Nvidia CUDA:cuda_9.0.176_win10
    Nvidia cuDNN:cudnn-9.0-windows10-x64-v7.4.2.24
    Tensorflow-gpu 版本:1.12.0
    Python 环境:3.6.4

    二、 安装 Tensorflow-gpu 包
    1. 请选择对应的 Tensorflow-gpu 版本,版本要求如下:

    https://tensorflow.google.cn/install/source_windows#gpu

    1. Anaconda 安装方式(推荐方式)(Anaconda 安装指南):
    conda install tensorflow-gpu=[版本号]
    
    1. pip 安装方式:
    pip install tensorflow-gpu=[版本号]
    
    1. 验证安装是否成功:
    import tensorflow as tf
    hello = tf.constant('hello world')
    world = tf.Session()
    print(world.run(hello))
    
    1. 查看 Tensorflow 版本和安装路径:
    print('Tensorflow 版本:',tf.__version__)
    print('Tensorflow 安装路径:',tf.__path__)
    

    相关文章

      网友评论

          本文标题:Tensorflow-GPU 环境安装与配置(Windows10

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