Windows下TensorFlow2.0 GPU环境配置

原创于 2020-10-17 13:52:36 发布 · 259 阅读

CC 4.0 BY-SA版权

文章标签：

本文介绍在Windows环境下配置TensorFlow2.0 GPU版本的具体步骤与注意事项，包括CUDA和cuDNN版本的选择，使用conda创建虚拟环境，以及安装特定版本的TensorFlow和其他依赖包。

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

经过实践，新手在TensorFlow2.0 GPU环境配置最容易弄错的地方就是各种包的版本之间对应问题，重要的事得说三遍。具体可网上寻找对应关系。

安装完cuda10.1和cudnn7.6.5后，接下来：

conda create -n tfgpu pip python=3.6.11；
进入虚拟环境运行pip install --ignore-installed --upgrade tensorflow_gpu==2.2.0；
如何需要openc-python则继续pip install opencv_python-4.4.0-cp36-cp36m-win_amd64.whl，注意此文件需下载，然后当前目录下去运行；
完成安装，运行测试代码。