tensorflow安装

安装 tensorflow-gpu 过程挺曲折的:
首先我想安装的是 tensorflow-gpu 1.14 基于pthon3的版本。
我安装tensorflow主要用于学习和研究,主要工作还是从事其它软件的开发工作,故而首先安装的是vs2017 .net 和 c++.
正确的步如下:
1. 安装 anaconda;
(1)新建一个环境 gtensorflow
>>conda create -n gtensorflow python=3.7
最初创建的时候写的3.5,但后面看到系统安装的是3.7
(2)激活环境
>>activate gtensorflow
(3)安装tensorflow-gpu
>>pip intsall tensorflow-gpu==1.14
2. 安装完后,去目录“C:\Users\aliange\Anaconda3\envs\gtensorflow\Lib\site-packages\tensorflow\python\platform”下打开 build_info.py 可以看到

cuda_version_number = '10.0' #cuda toolkits 的版本
cudnn_version_number = '7' #cudnn的大版本
msvcp_dll_name = 'msvcp140.dll' #这个和vc相关是编译时使用的vc版本
nvcuda_dll_name = 'nvcuda.dll'
cudart_dll_name = 'cudart64_100.dll'
cudnn_dll_name = 'cudnn64_7.dll' #这个和cudnn是息息相关的,下载完后可以看看bin目录下的库是不是和这名称相同

3. 根据第2步去选择英伟达官网找对应的驱动,把cudnn解压后放到 “C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0”下

4. 此时 测试未通过,首先卸载 tensorflow-gpu 然后重新安装即可。


posted @ 2019-10-14 18:17  神棍二叔  阅读(175)  评论(0编辑  收藏  举报