Cargo仓库中TensorFlow GPU构建脚本修改无效的解决办法

问题根源 #

你使用的tensorflow crate 0.20.0依赖tensorflow-sys 0.23.0，其构建脚本硬编码了TensorFlow版本为2.11.0，但Windows平台的GPU版TensorFlow 2.11.0并未发布，导致下载不存在的压缩包引发解压错误。直接修改本地registry中的build.rs会被Cargo覆盖，无法生效。

解决方案（无需降级TensorFlow crate） #

方案1：手动指定本地预下载的GPU库

无需修改依赖源码，通过环境变量让构建脚本跳过下载步骤，直接使用本地已有的正确版本库：

• 下载Windows平台GPU版TensorFlow 2.10.0的预编译库libtensorflow-gpu-windows-x86_64-2.10.0.zip
• 将压缩包解压到任意目录，例如C:\tools\libtensorflow-gpu-2.10.0
• 设置环境变量：

  set TENSORFLOW_LIB_DIR=C:\tools\libtensorflow-gpu-2.10.0\lib
  set TENSORFLOW_INCLUDE_DIR=C:\tools\libtensorflow-gpu-2.10.0\include
  

• 执行cargo build，构建脚本会自动检测到环境变量，跳过下载流程，直接使用本地库完成构建。

方案2：通过Cargo Patch修改tensorflow-sys的构建脚本

如果需要长期修复这个问题，可以通过Cargo的patch功能替换修改后的tensorflow-sys源码：

1. 从crates.io下载tensorflow-sys 0.23.0的源码包，或从GitHub拉取对应tag的代码
2. 修改源码中的build.rs文件：
   • 找到VERSION常量，将其改为"2.10.0"；或者针对GPU场景单独处理版本，比如：

     let version = if proc_type == "gpu" && os == "windows" {
         "2.10.0"
     } else {
         VERSION
     };
     

     然后在生成下载链接时使用这个version变量替代原有的VERSION常量
3. 在你的项目Cargo.toml中添加patch配置，指向修改后的本地tensorflow-sys目录：

   [patch.crates-io]
   tensorflow-sys = { path = "./path/to/your/modified/tensorflow-sys" }
   

4. 执行cargo build，Cargo会使用你本地修改后的tensorflow-sys进行构建。

内容的提问来源于stack exchange，提问作者user210366