机器学习之路-win10下tensorflow-gpu环境的安装与配置

win10下tensorflow-gpu(2.x版本)环境的安装与配置

  • Anaconda的安装
  • 安装Python3.6
  • GPU版本所需的额外软件-CUDA
    • 配置环境变量-CUDA
  • GPU版本所需的额外软件-cuDNN
  • Tensorflow安装
  • Tensorflow测试
    • 错误1:在xxx路径(CUDA安装目录下)下没有找到xxx.dll(如cudnn64_7.dll)
    • 错误2:RuntimeError

参考:
https://www.cnblogs.com/tan-wm/p/9566564.html
https://blog.csdn.net/weixin_41655940/article/details/103570512

从环境配置开始记录一下机器学习的过程,也是第一篇博客,希望能帮助到看到的人
电脑环境:windows10专业版+RTX2070S
安装软件:
Anaconda3 5.2.0
Tensorflow 2.2.0 gpu版
CUDA 10.2
cuDNN 7.6.5

Anaconda的安装

为不能的朋友:清华源
需要最新版的朋友:官网

下载完成后运行,注意勾选 add anaconda to my PATH environment variable
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第1张图片
中间有Visual Studio integration安装,经常报错,建议不要安装(错误内容:网络链接不稳定)

安装Python3.6

为了保证稳定性,选择安装python3.6
打开cmd,使用命令

conda create -n tensorflow pip python=3.6

机器学习之路-win10下tensorflow-gpu环境的安装与配置_第2张图片
从提示中得知,要使用这个环境,需要命令 activate tensorflow
在这里插入图片描述
出现如上提示,即为成功

GPU版本所需的额外软件-CUDA

1.CUDA:CUDA Toolkit
点击查看CUDA与显卡驱动对应关系(如下图)
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第3张图片
自行选择与显卡驱动版本相适应的CUDA版本
笔者选择network版本下Base Installer安装:
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第4张图片
此处network版本较小,但是后续使用需要联网,local版本较大,但是不需联网,请根据自身需要选择。

机器学习之路-win10下tensorflow-gpu环境的安装与配置_第5张图片
(RTX系列安装CUDA9.0会出现该提示,建议更换10.0)

正常开始安装后,第一次安装建议全选,但是如果选择Visual Studio integration会出现上文中出现过的网络连接错误
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第6张图片
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第7张图片
记住路径,以便于配置环境变量,接下来开始安装等待…
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第8张图片

配置环境变量-CUDA

在 “我的电脑-属性-高级系统设置-环境变量” 中找到path,添加以下环境变量(cuda使用默认安装路径):
“系统变量”中 出现CUDA_PATH,NVCUDASAMPLES_ROOT相关内容
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第9张图片
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第10张图片编辑path变量
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第11张图片
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp
新加入进去
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第12张图片
后在CMD中输入 nvcc -V指令,跳出版本信息,则CUDA安装成功
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第13张图片

GPU版本所需的额外软件-cuDNN

cuDNN官网(需登入)

选择对应版本 笔者下载的是CUDA10.2,故选择cuDNN v7.6.5
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第14张图片
将下载文件解压,得到
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第15张图片
全选,复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2(上文CUDA安装地址)中
复制后:
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第16张图片

Tensorflow安装

参考Tensorflow官网,查看CPU,编译环境与Tensorflow的对应关系(如下图)
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第17张图片
选择好对应的Tensorflow版本后,进行下载

法1(默认下载最新版,确认好版本对应之后可用):
注意使用管理员权限进入
Anaconda prompt中输入命令

activate tensorflow
pip install --user --ignore-installed --upgrade tensorflow-gpu

机器学习之路-win10下tensorflow-gpu环境的安装与配置_第18张图片

法2
或输入命令

activate tensorflow
conda install tensorflow-gpu

(此处图片用Anaconda prompt做演示)

机器学习之路-win10下tensorflow-gpu环境的安装与配置_第19张图片
机器学习之路-win10下tensorflow-gpu环境的安装与配置_第20张图片
y/n选项选择y

如果出现timeout错误,请尝试使用VPN

法3 :清华源链接下载
选择好对应的版本,点击即可下载,下载后在cmd中进入下载根目录,输入

pip install tensorflow对应的文件名.whl

等待安装完成即可

Tensorflow测试

使用代码

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.compat.v1.Session()
print(sess.run(hello))

中途没有报错信息,成功出现下图字符,则成功
在这里插入图片描述

错误1:在xxx路径(CUDA安装目录下)下没有找到xxx.dll(如cudnn64_7.dll)

在Tensorflow的官网给出的安装cuDNN的操作中,只是将cuDNN的文件解压到如C:\tools\cuda,并不需要把cudnn的文件拷贝到CUDA的安装目录中,去检查CUDA的这几个文件夹,将cuDNN对应文件拷贝覆盖进入解决

错误2:RuntimeError

机器学习之路-win10下tensorflow-gpu环境的安装与配置_第21张图片
使用Tensorflow 2.x版本时,可能会出现该错误,原因是其版本不兼容
将测试代码改为:

import tensorflow as tf
tf.compat.v1.disable_eager_execution() #添加一行
hello = tf.constant('Hello, TensorFlow!')
sess = tf.compat.v1.Session()
print(sess.run(hello))

即可

你可能感兴趣的:(机器学习之路-win10下tensorflow-gpu环境的安装与配置)