如何通过requirements.txt安装指定Tensorflow wheel?含安装与打包故障
解决TensorFlow定制Wheel包的安装与requirements.txt配置问题
一、直接用Git链接安装失败的原因
你之前用的pip install "git+ssh://git@github.com/fo40225/tensorflow-windows-wheel/tree/master/2.9.0/py39/CPU%2BGPU/cuda117cudnn8sse2"命令报错,是因为这个链接是GitHub网页的目录地址,不是Git仓库的可克隆路径,也不是pip能识别的合法包安装源。pip的Git安装方式仅支持指向仓库根目录或包含setup.py的源码子目录,而我们需要的是已经编译好的wheel包,所以这种方式行不通。
二、无需自行打包wheel文件(现成包已存在)
你之前尝试wheel pack tensorflow报错No .dist-info directories found in tensorflow,是因为你下载的是源码目录,缺少wheel打包必需的元数据目录。但实际上那个GitHub目录下已经提供了编译好的适配老旧CPU(SSE2)的wheel包,完全不需要自己打包,直接用现成的即可。
三、两种可行的安装与requirements.txt配置方案
方案1:本地下载wheel包后使用
- 打开指定的GitHub目录,找到对应Python3.9的
.whl文件(比如tensorflow-2.9.0-cp39-cp39-win_amd64.whl),下载到你的项目目录下(可以新建一个wheels子目录存放,方便管理)。 - 执行安装命令:
pip install ./wheels/tensorflow-2.9.0-cp39-cp39-win_amd64.whl - 在
requirements.txt中添加以下内容(路径根据你存放的位置调整):./wheels/tensorflow-2.9.0-cp39-cp39-win_amd64.whl
方案2:通过Raw URL直接在线安装
如果不想本地保存wheel文件,可以使用该包的Raw链接(找到wheel文件后点击页面上的「Raw」按钮获取链接):
- 安装命令:
pip install "https://raw.githubusercontent.com/fo40225/tensorflow-windows-wheel/master/2.9.0/py39/CPU%2BGPU/cuda117cudnn8sse2/tensorflow-2.9.0-cp39-cp39-win_amd64.whl" - 在
requirements.txt中直接添加该URL:https://raw.githubusercontent.com/fo40225/tensorflow-windows-wheel/master/2.9.0/py39/CPU%2BGPU/cuda117cudnn8sse2/tensorflow-2.9.0-cp39-cp39-win_amd64.whl
四、关于自行打包的补充(不推荐)
如果确实需要自行编译打包,你需要从该仓库获取完整的构建配置和适配SSE2的源码补丁,然后按照TensorFlow的官方构建流程操作,但这步骤繁琐且容易出错。既然仓库已经提供了现成的适配包,优先使用现成方案更高效。
内容的提问来源于stack exchange,提问作者Juan C
相关产品推荐
相关产品推荐

