1. 为什么你需要这份指南?
每次在Windows上配置TensorFlow GPU环境都像在拆炸弹——剪错一根线就前功尽弃。去年我帮学弟调试毕设时,发现他照着某视频教程安装的TensorFlow 2.0 + CUDA 10.0组合根本跑不动YOLOv5,显卡利用率始终为0%。这不是个例,网上80%的教程都存在版本匹配问题。
这里有个血泪教训:TensorFlow对CUDA/cuDNN的版本要求极其严格。比如TensorFlow 2.8.0官方明确要求CUDA 11.2和cuDNN 8.1,但很多教程还在推荐CUDA 10.1。更坑的是,NVIDIA驱动版本必须≥450.80.02,这个细节90%的教程都会漏掉。
我的GTX 1080Ti就曾因此变成"高级暖气片"——训练时GPU使用率不到5%。后来发现是CUDA Toolkit安装时漏勾了Visual Studio集成选项。这份指南会带你避开所有暗坑,从驱动版本校验到环境变量配置,每个环节都有双重验证机制。
2. 基础环境搭建
2.1 Miniconda的精准安装
别再用Anaconda了!动辄3GB的安装包会塞满你的C盘。Miniconda只有80MB,却能实现同样的环境隔离效果。下载时要注意两个致命细节:
- 必须选择与Python 3.8对应的版本(比如Miniconda3-py38_4.12.0-Windows-x86_64.exe)
- 安装时要勾选"Add to PATH"(否则后面调用conda命令会报错)
验证安装成功的正确姿势不是简单运行conda --version,而是用这个组合命令:
conda create -n tf_gpu python=3.8 && conda activate tf_gpu
python -c "import sys; print(sys.executable)"
这能同时测试conda环境创建和Python解释器路径是否正确。</


1022

被折叠的 条评论
为什么被折叠?



