平时使用requirements.txt
文件来安装python
的依赖,如下所示:
Flask==3.0.0
Flask-Cors==4.0.0
elastic-transport==8.11.0
elasticsearch==8.11.1
但是如果我们的依赖中包含pytorch
依赖,显然是不能简单的通过这个方式来进行的,例如:
Flask==3.0.0
Flask-Cors==4.0.0
elastic-transport==8.11.0
elasticsearch==8.11.1
torch==2.1.2+cu118
torchaudio==2.1.2+cu118
torchvision==0.16.2+cu118
使用这个requirements.txt
安装会报错:
解决办法
在requirements.txt
的顶部加入--find-links https://download.pytorch.org/whl/torch_stable.html
,如下:
--find-links https://download.pytorch.org/whl/torch_stable.html
Flask==3.0.0
Flask-Cors==4.0.0
elastic-transport==8.11.0
elasticsearch==8.11.1
torch==2.1.2+cu118
torchaudio==2.1.2+cu118
torchvision==0.16.2+cu118
这样在执行pip install ./requirements.txt
就没有问题了