主要介绍CUDA与TensorFlow的环境搭建配置的大体思路，具体细节根据自己情况调整！

（1）Windows查看CUDA版本号

详情具体见：Windows系统查看CUDA版本号

原因：在配置Tensorflow时要求与CUDA版本相对应，本文配置主要是CUDA 11. 因此，弄清CUDA版本号很重要。

（2）win10环境下CUDA和CUDNN的安装

具体参考：win10下CUDA和CUDNN的安装（超详细）！亲测有效！

                 Win10+Tensorflow2 + CUDA11 GPU环境搭建

另外，tensorflow-gpu版本国内镜像源：tensorflow-gpu 清华镜像源

（3）TensorFlow与CUDA对应版本号参考，选择安装对应的TensorFlow版本(GPU).

网址：https://tensorflow.google.cn/install/source_windows

示例参考教程（具体硬件要求）：

RTX2080Ti+Win10+Cuda10.1+Cudnn7.6+tensorflow-gpu2

特别注意：用conda安装仅限于当前环境使用，而且会自动安装相应的cudnn和cuda支持，不需要手动安装，因此建议使用conda安装相应的TensorFlow版本。

转载本文请联系原作者获取授权，同时请注明本文来自张伟科学网博客。

链接地址：https://m.sciencenet.cn/blog-3428464-1308544.html?mobile=1

收藏

分享