Azure部署Flask应用时如何安装指定URL的PyTorch版本？

1. 修改requirements.txt，指定PyTorch nightly源 #

直接在requirements.txt开头加上pip安装所需的参数，再列出依赖包，这样不管是本地还是Azure部署，pip都会从指定的nightly CUDA源拉取兼容Python3.12的预发布版本：

--pre
--index-url https://download.pytorch.org/whl/nightly/cu118
torch
torchvision
torchaudio
# 下面放你的其他依赖，比如Flask等
flask
# ...其他项目依赖

本地测试时直接运行pip install -r requirements.txt就能正常安装，Azure部署时也会自动读取这些参数拉取对应版本。

2. Azure环境适配（若仍有问题） #

如果Azure用的是普通CPU实例，即使安装成功也没法用CUDA，得切换到GPU实例（比如Azure App Service Premium v3 GPU层、GPU虚拟机或GPU容器实例）。另外也可以在部署脚本里显式执行安装命令：
比如在Azure的启动脚本（如startup.sh）里先安装PyTorch，再启动Flask：

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118
gunicorn app:app

3. 本地报错的原因及解决 #

你本地用自动生成的requirements.txt安装失败，是因为自动生成的文件只写了torch==x.x.x+cu118这类版本号，但没指定源——PyPI官方仓库里没有这些nightly预发布版本，所以必须用上面带源的requirements.txt格式。

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