首页 » 深度学习 » 正文

TensorFlow CUDA cuDNN配置相关问题

解决自己在安装TensorFlow CUDA cuDNN时遇到的问题

版本问题

Win10 + 1060 目前使用TensorFlow1.12 CUDA9.0没有出现问题(显卡驱动是最新的)。
在这个网址上可以查看到官方给出的TensorFlow和CUDA的版本支持列表,但是官方只写到TensorFlow1.8版本。
https://www.tensorflow.org/install/source#common_installation_problems
亲测Win10 + 1060使用TensorFlow1.12 CUDA9.0是没有问题的。

步骤

  1. 首先更新显卡驱动,目前Win10也会自动更新显卡驱动到稳定版本,基本不用担心这个出错的问题。
  2. 下载CUDA对应版本,并且根据版本选择对应的cuDNN下载。
  3. 安装CUDA,建议选择自定义安装,只需安装CUDA部分,若已经有VS环境可以把VS安装也取消勾选。
  4. CUDA安装完成后,解压cuDNN文件,将对应目录的文件拷贝到CUDA安装目录的对应文件夹中。
  5. 添加环境变量,需要添加以下三个环境变量,具体位置根据安装路径确定。
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64;%PATH%
SET PATH=C:\tools\cuda\bin;%PATH%

其他

强烈建议阅读官方文档,网上的安装教程(包括本篇)都会存在版本问题,导致安装不成功。
再强调一遍,重中之重就是要确保TensorFlow CUDA cuDNN三者的版本是统一的。

赞 (2)

发表评论