Win10+Python3.6搭建Tensorflow-gpu环境

前言

楼主装这个环境装了N遍,踩了不少坑,之前曾经安装好了,但是自己也不懂,搞了N遍总算搞清楚了。我也看了不少文章,怎么安装,但是总有一些是没写清楚,现在把我自己装的过程发出来给大家看。希望能帮到你们,楼主是Win10 64位的系统在Anaconda下配置的tensorflow gpu的环境,第一次写,写的不好敬请见谅

常见的问题

Tensorflow GPU版的安装,有下面几个问题一定要注意:

  1. Python版本和TensorFlow GPU的版本一定要对应
  2. TensorFlow GPU和CUDA的版本要对应
  3. CUDA的版本和VS的版本要对应
  4. 装CUDA前要先装了VS;
  5. 版本参考:https://tensorflow.google.cn/install/source_windows

所需工具

Anaconda https://www.anaconda.com/download/ 楼主下载的是3.5

VS下载 https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/
CuDa https://developer.nvidia.com/cuda-toolkit-archive
cudnn https://developer.nvidia.com/cudnn
tensorflow https://pypi.org/project/tensorflow-gpu/1.12.0/#files

安装顺序

先安装Anaconda 过程不详细说
到这一步的时候,不选,环境自己去配
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
安装完成后,配置环境变量

Path里面将以下三个路径加入,注意这里要换成你自己的安装路径。
C:\Users\t-yaoguo\AppData\Local\Continuum\anaconda3
C:\Users\t-yaoguo\AppData\Local\Continuum\anaconda3\Scripts
C:\Users\t-yaoguo\AppData\Local\Continuum\anaconda3\Library\bin

配置完成后 Anaconda就可以使用了
在CMD中输入 “conda -V”,能正常显示版本号,证明已经配置好了。
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
装CUDA之前确保已经装好对应的VS 如果电脑上没有VS,请先装了VS
CUDA的一些功能需要VS的C++环境才能使用,若没装VS就直接装CUDA会造成部分功能无法使用,遇到的问题下面再说。

安装CUDA
在这里,楼主选择的是精简,自定义和精简安装的区别,楼主还不确定会不会有影响,之前参考的有精简有自定义,后面把自定义安装成功的链接贴上
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
等待安装完成之后,把cudnn的文件复制过去,如下:
把cudnn解压出来的bin 、include 、lib文件夹下的文件分别复制到cuda安装目录下对应的bin、include、lib文件里面
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
配置环境
依此添加下列三个路径到Path里(把路径换成自己的安装路径)

D:\NVIDIA\CUDAv9.0\bin
D:\NVIDIA\CUDAv9.0\include
D:\NVIDIA\CUDAv9.0\lib\x64

这样cuda和cudnn的安装就完成了

测试:在 CMD 中输入 nvcc -V,如果输出是 CUDA 版本信息,则说明安装成功
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
TensorFlow GPU安装
打开Anaconda Prompt 输入以下命令创建conda的环境

conda create -n tensorflow-gpu pip python=3.6
然后到这一步的时候输入 y 就会自动创建一个名为 tensorflow-gpu的环境,会下载这个环境下所需要的东西
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
上面的###是下载的东西
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
完成后**环境
输入 activate tensorflow-gpu
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
然后会变成这样
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
在这里输入 pip install 然后把你下载好的tensorflow_gpu1.12的whl文件拉进来,然后回车,就会自动安装

安装完成的图是类似这样
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建
输入python 测试是否成功
依次输入

import tensorflow as tf

hello = tf.constant(“Hello,TensorFlow!”)
sess = tf.Session()

print(sess.run(hello))

sess.close()
如下图就是安装成功了
Win10+Python3.6+Cuda9.0 Tensorflow1.12 Anaconda环境搭建

楼主遇到过的问题

Failed to load the native TensorFlow runtime.
这个问题据我所知是cuda安装的时候有问题

pip install tensorflow 会直接下载最新版的

cuda安装后再卸载好像卸载不干净

还有其他的暂时想不起来,以后遇到了再补上

相关文章:

  • 2022-02-26
  • 2021-07-10
  • 2022-12-23
  • 2022-12-23
  • 2021-07-21
  • 2021-10-31
  • 2021-04-26
猜你喜欢
  • 2022-12-23
  • 2021-07-13
  • 2021-11-30
  • 2021-12-01
  • 2021-10-23
  • 2021-05-21
  • 2021-07-06
相关资源
相似解决方案