前言
楼主装这个环境装了N遍,踩了不少坑,之前曾经安装好了,但是自己也不懂,搞了N遍总算搞清楚了。我也看了不少文章,怎么安装,但是总有一些是没写清楚,现在把我自己装的过程发出来给大家看。希望能帮到你们,楼主是Win10 64位的系统在Anaconda下配置的tensorflow gpu的环境,第一次写,写的不好敬请见谅
常见的问题
Tensorflow GPU版的安装,有下面几个问题一定要注意:
- Python版本和TensorFlow GPU的版本一定要对应;
- TensorFlow GPU和CUDA的版本要对应 ;
- CUDA的版本和VS的版本要对应;
- 装CUDA前要先装了VS;
- 版本参考: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 过程不详细说
到这一步的时候,不选,环境自己去配
安装完成后,配置环境变量
在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”,能正常显示版本号,证明已经配置好了。
装CUDA之前确保已经装好对应的VS 如果电脑上没有VS,请先装了VS
CUDA的一些功能需要VS的C++环境才能使用,若没装VS就直接装CUDA会造成部分功能无法使用,遇到的问题下面再说。
安装CUDA
在这里,楼主选择的是精简,自定义和精简安装的区别,楼主还不确定会不会有影响,之前参考的有精简有自定义,后面把自定义安装成功的链接贴上
等待安装完成之后,把cudnn的文件复制过去,如下:
把cudnn解压出来的bin 、include 、lib文件夹下的文件分别复制到cuda安装目录下对应的bin、include、lib文件里面
配置环境
依此添加下列三个路径到Path里(把路径换成自己的安装路径)
D:\NVIDIA\CUDAv9.0\bin
D:\NVIDIA\CUDAv9.0\include
D:\NVIDIA\CUDAv9.0\lib\x64
这样cuda和cudnn的安装就完成了
测试:在 CMD 中输入 nvcc -V,如果输出是 CUDA 版本信息,则说明安装成功
TensorFlow GPU安装
打开Anaconda Prompt 输入以下命令创建conda的环境
conda create -n tensorflow-gpu pip python=3.6
然后到这一步的时候输入 y 就会自动创建一个名为 tensorflow-gpu的环境,会下载这个环境下所需要的东西
上面的###是下载的东西
完成后激活环境
输入 activate tensorflow-gpu
然后会变成这样
在这里输入 pip install 然后把你下载好的tensorflow_gpu1.12的whl文件拉进来,然后回车,就会自动安装
安装完成的图是类似这样
输入python 测试是否成功
依次输入
import tensorflow as tf
hello = tf.constant(“Hello,TensorFlow!”)
sess = tf.Session()
print(sess.run(hello))
sess.close()
如下图就是安装成功了
楼主遇到过的问题
Failed to load the native TensorFlow runtime.
这个问题据我所知是cuda安装的时候有问题
pip install tensorflow 会直接下载最新版的
cuda安装后再卸载好像卸载不干净
还有其他的暂时想不起来,以后遇到了再补上