pytorchCU124不支持50系显卡

[Simonqujian78]

DockerFile 中：RUN pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 -i https://download.pytorch.org/whl/cu124 版本不支持50系显卡，需要更新到nightly，如何修改Dockerfile.

[zenosai]

如果需要安装nightly版本的pytorch的话，可以先注释掉

MonkeyOCR/docker/Dockerfile

Line 24 in acbd656

RUN pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 -i https://download.pytorch.org/whl/cu124

这一行，再在

MonkeyOCR/docker/Dockerfile

Line 36 in acbd656

RUN pip install -e .

这一行之后添加
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126 （CUDA 12.6）
或者
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118 （CUDA 11.8）
注意，我们没有测试过版本兼容性，无法确保可靠性。

欢迎随时反馈~

[Simonqujian78]

目前只有cu128的nightly支持50系显卡

[zenosai]

那就添加 pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

[Simonqujian78]

好的，晚上回家试试，及时反馈进展。

那就添加 pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

[Simonqujian78]

那就添加 pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

部署成功，运行成功，就是显存和内存全部占满，32+64，10页，大约20秒，格式不错。

[zhukefucn]

那就添加 pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

部署成功，运行成功，就是显存和内存全部占满，32+64，10页，大约20秒，格式不错。

请教老师了，我完全按照您说的做的，但是还是没搞定，能否再详细点讲讲啊。

[Simonqujian78]

具体遇到什么问题？Docker部署吗？

那就添加 pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

部署成功，运行成功，就是显存和内存全部占满，32+64，10页，大约20秒，格式不错。

请教老师了，我完全按照您说的做的，但是还是没搞定，能否再详细点讲讲啊。