windows下搭建tensorflow开发环境

最新推荐文章于 2023-10-03 16:50:10 发布

原创最新推荐文章于 2023-10-03 16:50:10 发布 · 471 阅读

0 ·

CC 4.0 BY-SA版权

本文详细介绍了如何在Windows环境下为TensorFlow 1.15配置GPU支持，包括CUDA 10.0和cuDNN v7.6.5的安装步骤，Anaconda环境的创建及TensorFlow的正确安装方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 安装cuda 10.0

下载地址：https://developer.nvidia.com/cuda-toolkit-archive

参考文档：https://blog.youkuaiyun.com/weixin_44307764/article/details/94909104

https://blog.youkuaiyun.com/weixin_45023983/article/details/99178625

cuda版本跟tensorflow版本要对应。 tensorflow1.15 对应cud10

2 cuDNN下载 Download cuDNN v7.6.5 (November 18th, 2019), for CUDA 10.2

下载深度神经网络加速库。版本必须与cuda对应。

https://developer.nvidia.com/rdp/cudnn-download

下载后将文件目录，拷贝到cuda安装目录：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\ + cuDNN\bin..

3 安装anaconda。

https://repo.anaconda.com/archive/Anaconda3-2019.10-Windows-x86_64.exe

安装简单，完成后cmd, 输入python 或ipython测试。

4 Anaconda下创建tensorflow环境。

参考：https://blog.youkuaiyun.com/XunCiy/article/details/89069193

Anaconda默认环境root环境下使用的是python3.7, 目前 tensorflow最新支持的python为3.6，

创建一个tensorflow的环境，使用python版本是3.6.5

conda create --name tensorflow python=3.6.5

安装完成后可以看到base环境下多了一个环境tensorflow

5 安装tensorflow

打开anaconda prompt命令行,请确保输入activate tensorflow后进入了tensorflow环境

安装指定GPU版本：pip install --ignore-installed --upgrade tensorflow-gpu==1.15，必须制定版本，否则安装最新版本，容易出现兼容性问题

下属对应图，实际测试，不是很准确，基本在附近。还是要实测。

Path	Compiler	CUDA/cuDNN	SIMD	Notes
2.0.0\py37\CPU\sse2	VS2019 16.3	No	x86_64	Python 3.7
2.0.0\py37\CPU\avx2	VS2019 16.3	No	AVX2	Python 3.7
2.0.0\py37\GPU\cuda101cudnn76sse2	VS2019 16.3	10.1.243_426.00/7.6.4.38	x86_64	Python 3.7/Compute 3.0
2.0.0\py37\GPU\cuda101cudnn76avx2	VS2019 16.3	10.1.243_426.00/7.6.4.38	AVX2	Python 3.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.15.0\py37\CPU+GPU\cuda101cudnn76sse2	VS2019 16.3	10.1.243_426.00/7.6.4.38	x86_64	Python 3.7/Compute 3.0
1.15.0\py37\CPU+GPU\cuda101cudnn76avx2	VS2019 16.3	10.1.243_426.00/7.6.4.38	AVX2	Python 3.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.14.0\py37\CPU\sse2	VS2019 16.1	No	x86_64	Python 3.7
1.14.0\py37\CPU\avx2	VS2019 16.1	No	AVX2	Python 3.7
1.14.0\py37\GPU\cuda101cudnn76sse2	VS2019 16.1	10.1.168_425.25/7.6.0.64	x86_64	Python 3.7/Compute 3.0
1.14.0\py37\GPU\cuda101cudnn76avx2	VS2019 16.1	10.1.168_425.25/7.6.0.64	AVX2	Python 3.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.13.1\py37\CPU\sse2	VS2017 15.9	No	x86_64	Python 3.7
1.13.1\py37\CPU\avx2	VS2017 15.9	No	AVX2	Python 3.7
1.13.1\py37\GPU\cuda101cudnn75sse2	VS2017 15.9	10.1.105_418.96/7.5.0.56	x86_64	Python 3.7/Compute 3.0
1.13.1\py37\GPU\cuda101cudnn75avx2	VS2017 15.9	10.1.105_418.96/7.5.0.56	AVX2	Python 3.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.12.0\py36\CPU\sse2	VS2017 15.8	No	x86_64	Python 3.6
1.12.0\py36\CPU\avx2	VS2017 15.8	No	AVX2	Python 3.6
1.12.0\py36\GPU\cuda100cudnn73sse2	VS2017 15.8	10.0.130_411.31/7.3.1.20	x86_64	Python 3.6/Compute 3.0
1.12.0\py36\GPU\cuda100cudnn73avx2	VS2017 15.8	10.0.130_411.31/7.3.1.20	AVX2	Python 3.6/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.12.0\py37\CPU\sse2	VS2017 15.8	No	x86_64	Python 3.7
1.12.0\py37\CPU\avx2	VS2017 15.8	No	AVX2	Python 3.7
1.12.0\py37\GPU\cuda100cudnn73sse2	VS2017 15.8	10.0.130_411.31/7.3.1.20	x86_64	Python 3.7/Compute 3.0
1.12.0\py37\GPU\cuda100cudnn73avx2	VS2017 15.8	10.0.130_411.31/7.3.1.20	AVX2	Python 3.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.11.0\py36\CPU\sse2	VS2017 15.8	No	x86_64	Python 3.6
1.11.0\py36\CPU\avx2	VS2017 15.8	No	AVX2	Python 3.6
1.11.0\py36\GPU\cuda100cudnn73sse2	VS2017 15.8	10.0.130_411.31/7.3.0.29	x86_64	Python 3.6/Compute 3.0
1.11.0\py36\GPU\cuda100cudnn73avx2	VS2017 15.8	10.0.130_411.31/7.3.0.29	AVX2	Python 3.6/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.11.0\py37\CPU\sse2	VS2017 15.8	No	x86_64	Python 3.7
1.11.0\py37\CPU\avx2	VS2017 15.8	No	AVX2	Python 3.7
1.11.0\py37\GPU\cuda100cudnn73sse2	VS2017 15.8	10.0.130_411.31/7.3.0.29	x86_64	Python 3.7/Compute 3.0
1.11.0\py37\GPU\cuda100cudnn73avx2	VS2017 15.8	10.0.130_411.31/7.3.0.29	AVX2	Python 3.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0,7.5
1.10.0\py36\CPU\sse2	VS2017 15.8	No	x86_64	Python 3.6
1.10.0\py36\CPU\avx2	VS2017 15.8	No	AVX2	Python 3.6
1.10.0\py36\GPU\cuda92cudnn72sse2	VS2017 15.8	9.2.148.1/7.2.1.38	x86_64	Python 3.6/Compute 3.0
1.10.0\py36\GPU\cuda92cudnn72avx2	VS2017 15.8	9.2.148.1/7.2.1.38	AVX2	Python 3.6/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.10.0\py27\CPU\sse2	VS2017 15.8	No	x86_64	Python 2.7
1.10.0\py27\CPU\avx2	VS2017 15.8	No	AVX2	Python 2.7
1.10.0\py27\GPU\cuda92cudnn72sse2	VS2017 15.8	9.2.148.1/7.2.1.38	x86_64	Python 2.7/Compute 3.0
1.10.0\py27\GPU\cuda92cudnn72avx2	VS2017 15.8	9.2.148.1/7.2.1.38	AVX2	Python 2.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.9.0\py36\CPU\sse2	VS2017 15.7	No	x86_64	Python 3.6
1.9.0\py36\CPU\avx2	VS2017 15.7	No	AVX2	Python 3.6
1.9.0\py36\GPU\cuda92cudnn71sse2	VS2017 15.7	9.2.148/7.1.4	x86_64	Python 3.6/Compute 3.0
1.9.0\py36\GPU\cuda92cudnn71avx2	VS2017 15.7	9.2.148/7.1.4	AVX2	Python 3.6/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.9.0\py27\CPU\sse2	VS2017 15.7	No	x86_64	Python 2.7
1.9.0\py27\CPU\avx2	VS2017 15.7	No	AVX2	Python 2.7
1.9.0\py27\GPU\cuda92cudnn71sse2	VS2017 15.7	9.2.148/7.1.4	x86_64	Python 2.7/Compute 3.0
1.9.0\py27\GPU\cuda92cudnn71avx2	VS2017 15.7	9.2.148/7.1.4	AVX2	Python 2.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.8.0\py36\CPU\sse2	VS2017 15.4	No	x86_64	Python 3.6
1.8.0\py36\CPU\avx2	VS2017 15.4	No	AVX2	Python 3.6
1.8.0\py36\GPU\cuda91cudnn71sse2	VS2017 15.4	9.1.85.3/7.1.3	x86_64	Python 3.6/Compute 3.0
1.8.0\py36\GPU\cuda91cudnn71avx2	VS2017 15.4	9.1.85.3/7.1.3	AVX2	Python 3.6/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.8.0\py27\CPU\sse2	VS2017 15.4	No	x86_64	Python 2.7
1.8.0\py27\CPU\avx2	VS2017 15.4	No	AVX2	Python 2.7
1.8.0\py27\GPU\cuda91cudnn71sse2	VS2017 15.4	9.1.85.3/7.1.3	x86_64	Python 2.7/Compute 3.0
1.8.0\py27\GPU\cuda91cudnn71avx2	VS2017 15.4	9.1.85.3/7.1.3	AVX2	Python 2.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.7.0\py36\CPU\sse2	VS2017 15.4	No	x86_64	Python 3.6
1.7.0\py36\CPU\avx2	VS2017 15.4	No	AVX2	Python 3.6
1.7.0\py36\GPU\cuda91cudnn71sse2	VS2017 15.4	9.1.85.3/7.1.2	x86_64	Python 3.6/Compute 3.0
1.7.0\py36\GPU\cuda91cudnn71avx2	VS2017 15.4	9.1.85.3/7.1.2	AVX2	Python 3.6/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.7.0\py27\CPU\sse2	VS2017 15.4	No	x86_64	Python 2.7
1.7.0\py27\CPU\avx2	VS2017 15.4	No	AVX2	Python 2.7
1.7.0\py27\GPU\cuda91cudnn71sse2	VS2017 15.4	9.1.85.3/7.1.2	x86_64	Python 2.7/Compute 3.0
1.7.0\py27\GPU\cuda91cudnn71avx2	VS2017 15.4	9.1.85.3/7.1.2	AVX2	Python 2.7/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.6.0\py36\CPU\sse2	VS2017 15.4	No	x86_64	Python 3.6
1.6.0\py36\CPU\avx2	VS2017 15.4	No	AVX2	Python 3.6
1.6.0\py36\GPU\cuda91cudnn71sse2	VS2017 15.4	9.1.85.3/7.1.1	x86_64	Python 3.6/Compute 3.0
1.6.0\py36\GPU\cuda91cudnn71avx2	VS2017 15.4	9.1.85.3/7.1.1	AVX2	Python 3.6/Compute 3.0,3.5,5.0,5.2,6.1,7.0
1.6.0\py27\CPU\sse2	VS2017 15.4	No	x86_64	Python 2.7