当前位置:首页 > 操作系统 > Windows

Windows下安装TensorFlow-gpu

在之前需要安装CUDA Toolkit 8.0.
https://pan.baidu.com/s/1c2y8xIK和cuDNN v6 or v6.1。
一定要版本正确,否则会出现问题,在官网上有说明。

直接一直点击确认即可,安装后可以看到系统中多了CUDA_PATH和CUDA_PATH_V8_0两个环境变量,接下来要在系统中添加以下几个环境变量:   CUDA_SDK_PATH=C:ProgramDataNVIDIA CorporationCUDA Samplesv8.0   CUDA_LIB_PATH=%CUDA_PATH%libx64

   CUDA_BIN_PATH=%CUDA_PATH%bin
   CUDA_SDK_BIN_PATH=%CUDA_SDK_PATH%binwin64

   CUDA_SDK_LIB_PATH=%CUDA_SDK_PATH%commonlibx64
   然后,在系统变量PATH的末尾依次添加:
   %CUDA_LIB_PATH%  %CUDA_BIN_PATH%  %CUDA_SDK_LIB_PATH%  %CUDA_SDK_BIN_PATH%  说明:这里注意的是CUDA_SDK_PATH路径的位置,很多人去这个错误路径“C:Program FilesNVIDIA Corporation”下,发现找不到CUDA Samples文件夹。实际上是因为ProgramData是一个隐藏文件夹,需要设置【我的电脑】-【查看】-【显示隐藏文件】才能看到,因此这个才是正确路径“C:ProgramDataNVIDIA CorporationCUDA Samplesv8.0”。
  配置成功后,重启计算机使环境变量生效。

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


【说明】本文章由站长整理发布,文章内容不代表本站观点,如文中有侵权行为,请与本站客服联系(QQ:254677821)!

相关教程推荐

其他课程推荐