WIN10 安装Tensorlow GPU版本

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版本)