我已经安装了 pytorch 1.7.1 版本
RUN pip install torch==1.7.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
然后在安装 sentence-transformers 2.0.0 的时候,为什么还会自动下载一个 pytorch,怎么才能用已经安装的这个 pytorch
Collecting torch>=1.6.0
Downloading https://pypi.tuna.tsinghua.edu.cn/packages/d2/a9/b3cea4a97ffabd6639e71608814dbd08081e202e8ac9580250273c0541ff/torch-1.9.0-cp37-cp37m-manylinux1_x86_64.whl (831.4 MB)
1
Trim21 2021-07-03 18:34:38 +08:00 via Android
是不是 1.7.1+cpu 作为版本号不算高于 1.6.0 ?
|
2
Jirajine 2021-07-03 18:55:46 +08:00 via Android
用 poetry 管理依赖,有 lock 文件,不用担心版本问题。
|
3
bankroft 2021-07-03 19:04:47 +08:00
如果 dockerfile 有新安装 py 的话,注意一下有没有多的 pip
|