Windows下安装TensorFlow-gpu

在以前须要安装CUDA Toolkit 8.0.
https://pan.baidu.com/s/1c2y8xIK和cuDNN v6 or v6.1。
必定要版本正确,不然会出现问题,在官网上有说明。python

直接一直点击确认便可,安装后能够看到系统中多了CUDA_PATH和CUDA_PATH_V8_0两个环境变量,接下来要在系统中添加如下几个环境变量:   CUDA_SDK_PATH=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0   CUDA_LIB_PATH=%CUDA_PATH%\lib\x64

   CUDA_BIN_PATH=%CUDA_PATH%\bin
   CUDA_SDK_BIN_PATH=%CUDA_SDK_PATH%\bin\win64

   CUDA_SDK_LIB_PATH=%CUDA_SDK_PATH%\common\lib\x64
   而后,在系统变量PATH的末尾依次添加:
   %CUDA_LIB_PATH%  %CUDA_BIN_PATH%  %CUDA_SDK_LIB_PATH%  %CUDA_SDK_BIN_PATH%  说明:这里注意的是CUDA_SDK_PATH路径的位置,不少人去这个错误路径“C:\Program Files\NVIDIA Corporation”下,发现找不到CUDA Samples文件夹。其实是由于ProgramData是一个隐藏文件夹,须要设置【个人电脑】-【查看】-【显示隐藏文件】才能看到,所以这个才是正确路径“C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0”。
  配置成功后,重启计算机使环境变量生效。

conda create -n tensorflow python=3.5
activate tensorflow
pip install –ignore-installed –upgrade tensorflow-gpuide

相关文章
相关标签/搜索