unir/MastersThesis
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

gpu amd 64
Some checks failed
build_docker / essential (pull_request) Successful in 1s
Details
build_docker / build_cpu (pull_request) Successful in 7s
Details
build_docker / build_gpu (pull_request) Failing after 7m24s
Details

Browse Source
This commit is contained in:
Sergio Jimenez Jimenez
2026-01-17 11:10:13 +01:00
parent 9daa496ab6
commit 02f8500ce5
3 changed files with 29 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/paddle_ocr/README.md
View File

@@ -129,6 +129,8 @@ docker buildx build -f Dockerfile.cpu \
docker build -f Dockerfile.gpu -t paddle-ocr-api:gpu .
```
> **Note:** PaddlePaddle GPU 3.x packages are **not on PyPI**. The Dockerfile installs from PaddlePaddle's official CUDA index (`paddlepaddle.org.cn/packages/stable/cu126/`). This is handled automatically during build.
## Running
### CPU (Any machine)
@@ -448,3 +450,14 @@ Ensure NVIDIA Container Toolkit is installed:
nvidia-smi # Should work
docker run --rm --gpus all nvidia/cuda:12.0-base nvidia-smi # Should work
```
### PaddlePaddle GPU installation fails
PaddlePaddle 3.x GPU packages are **not available on PyPI**. They must be installed from PaddlePaddle's official index:
```bash
# For CUDA 12.x
pip install paddlepaddle-gpu==3.2.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
# For CUDA 11.8
pip install paddlepaddle-gpu==3.2.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
```
The Dockerfile.gpu handles this automatically.