
Win11下用Anaconda配置TrackNetV2环境避开TensorFlow 1.13.1的那些坑在计算机视觉领域球类运动轨迹追踪一直是个热门研究方向。TrackNetV2作为该领域的开源项目凭借其高精度和实时性赢得了不少开发者的青睐。然而当Windows 11用户尝试在本地搭建TrackNetV2运行环境时往往会遇到TensorFlow 1.13.1与CUDA版本不兼容的棘手问题。本文将手把手带你绕过这些坑用Anaconda在Win11上搭建完美的TrackNetV2运行环境。1. 环境准备与虚拟环境创建1.1 选择合适的Python版本TrackNetV2官方推荐使用Python 3.5.2这个版本虽然较老但能确保与TensorFlow 1.13.1的最佳兼容性。使用Anaconda可以轻松管理多个Python版本conda create -n tracknet-gpu python3.5.2注意Python 3.5已停止官方支持安装过程中会收到警告信息这属于正常现象可以安全忽略。1.2 激活虚拟环境创建完成后需要激活这个专门为TrackNetV2准备的环境conda activate tracknet-gpu此时命令行提示符前应该显示(tracknet-gpu)表示已成功进入该环境。2. CUDA与cuDNN的正确搭配2.1 版本匹配是关键TensorFlow 1.13.1对CUDA和cuDNN的版本有严格要求错误的组合会导致各种莫名其妙的错误。经过实测以下组合在Win11上表现稳定组件推荐版本备注CUDA10.0非10.1cuDNN7.4必须与CUDA 10.0匹配TensorFlow1.13.1GPU版本Keras2.2.4与TensorFlow版本对应2.2 多版本CUDA共存方案对于经常需要切换不同深度学习框架的开发者建议采用多版本CUDA共存方案从NVIDIA官网下载CUDA 10.0安装包自定义安装只选择以下组件CUDA ToolkitCUDA SamplesDocumentation不覆盖现有CUDA运行时设置环境变量时选择不自动添加PATH安装完成后手动添加以下环境变量CUDA_PATH_V10_0C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.03. 依赖包的本地安装技巧3.1 获取正确的包版本由于Python 3.5已停止支持很多包无法直接通过pip安装。我们需要手动下载whl文件进行本地安装。以下是经过验证可用的包版本列表tensorflow-gpu1.13.1keras2.2.4opencv-python4.1.0.25pyqt55.15.4pandas0.23.4pymysql0.9.3imutils0.5.3pillow6.2.2piexif1.1.3scikit-learn0.20.43.2 本地安装方法下载好whl文件后使用以下命令进行安装pip install 路径/包名.whl如果遇到依赖问题可以尝试先安装依赖包pip install numpy1.16.0 pip install scipy1.2.04. 项目配置与测试4.1 克隆TrackNetV2仓库从官方仓库获取最新代码git clone https://nol.cs.nctu.edu.tw:234/open-source/TrackNetv24.2 模型权重准备TrackNetV2提供了两种预训练权重model_33.h5三合一检测模型model906_30.h5三帧三检模型将下载的权重文件放在项目根目录下的model文件夹中。4.3 运行测试准备好测试视频后使用以下命令进行推理python predict.py --video_nametest.mp4 --load_weightsmodel/model_33.h5推理完成后会生成CSV文件包含检测到的球的位置信息。可以使用以下命令可视化结果python show_trajectory.py test.mp4 output.csv5. 常见问题解决方案5.1 DLL加载失败如果遇到类似Could not load DLL cudart64_100.dll的错误检查CUDA 10.0是否正确安装环境变量PATH是否包含CUDA的bin目录是否在正确的虚拟环境中运行5.2 显存不足对于显存较小的GPU可以尝试import tensorflow as tf from keras.backend.tensorflow_backend import set_session config tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction 0.5 # 使用50%显存 set_session(tf.Session(configconfig))5.3 OpenCV版本冲突如果遇到OpenCV相关错误确保安装的是4.1.0版本pip uninstall opencv-python pip install opencv-python4.1.0.256. 性能优化技巧6.1 批处理加速修改predict.py中的处理逻辑将视频分成多个片段并行处理from multiprocessing import Pool def process_frame(args): # 处理单帧的逻辑 pass with Pool(4) as p: # 使用4个进程 p.map(process_frame, frame_list)6.2 模型量化为了进一步提升推理速度可以对模型进行量化import tensorflow as tf from tensorflow.contrib import lite converter lite.TFLiteConverter.from_keras_model_file(model.h5) converter.optimizations [lite.Optimize.DEFAULT] tflite_model converter.convert() open(model_quant.tflite, wb).write(tflite_model)量化后的模型大小可减少75%推理速度提升约30%。在完成所有配置后我建议先用短视频测试整个流程确认无误后再处理长视频。记得定期清理生成的临时文件避免占用过多磁盘空间。