美文网首页TensorFlow操作
WIN10 安装Tensorlow GPU版本

WIN10 安装Tensorlow GPU版本

作者: 翻开日记 | 来源:发表于2018-11-28 19:26 被阅读0次

    Python 安装

    1. Anaconda 安装(注:安装进度完成后,勾选自动添加路径)
    2. 创建环境并激活环境
      conda create -n Gtensorflow pip python=3.5
      注:尽量使用pyhton3.5
      activate Gtensorlow
      注:可以不创建环境,直接在base环境中安装
    3. 使用pip安装Tensorlow GPU版
      pip install --ignore-installed --upgrade tensorflow-gpu==1.9 -i https://pypi.tuna.tsinghua.edu.cn/simple
      注:==1.9如果不写会默认安装低版本(1.1); -i使用清华源安装

    CUDA 安装

    1. 打开文件Anaconda3\envs\tensorflow\Lib\site-packages\tensorflow\python\platform\build_info.py看需要的版本
        cudart_dll_name = 'cudart64_90.dll'
        cuda_version_number = '9.0'
        nvcuda_dll_name = 'nvcuda.dll'
        cudnn_dll_name = 'cudnn64_7.dll'
        cudnn_version_number = '7'
    

    nvcuda.dllmsvcp140.dllC:\Windows\System32\中需要存在
    cuda_version_number = '9.0'cudnn_version_number = '7'表示需要cuda9.0版本,cudnn7.0+版本
    注:也可以查看TensorFlow官网:https://www.tensorflow.org/install/install_windows#requirements_to_run_tensorflow_with_gpu_support

    1. 在英伟达官网找对应版本的CUDA 和 cuDNN

    CUDA:https://developer.nvidia.com/cuda-toolkit-archive
    注意:win10 下需要下载主文件(1.4G)+多个补丁
    cuDNN:https://developer.nvidia.com/rdp/cudnn-archive

    1. 安装CUDA
      注意:(1)第一个大选项中,去掉勾选Visual Studio Integration,(2)去掉Driver components大选项(如果已经安装了N卡驱动,一般都安装了)(3)默认路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\

    2. 依次安装多个补丁包
      注:会存在补丁包打开后闪退,可忽略

    3. 解压cudnn-9.0-windows10-x64-v7.4.1.5.zip,把bin,include,lib里面的文件(共三个)分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\下的bin,include,lib相应目录中

    4. 在系统环境变量中PATH加入

    
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
    
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
    
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp
    

    测试

    查看是否使用GPU

    import tensorflow as tf
    tf.test.gpu_device_name()
    

    相关文章

      网友评论

        本文标题:WIN10 安装Tensorlow GPU版本

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