Skip to content

Commit 67c68d8

Browse files
committed
add troubleshooting notice to README.md
1 parent 5f10f91 commit 67c68d8

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

examples/image_search/README.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,20 @@ export OLLAMA_MODEL="gemma3" # Optional, for caption generation
7676
uvicorn colpali_main:app --reload --host 0.0.0.0 --port 8000
7777
```
7878

79+
Note that recent Nvidia GPUs (RTX 5090) will not work with the Stable pytorch version up to 2.7.1
80+
81+
If you get this error:
82+
83+
```
84+
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90 compute_37.
85+
```
86+
87+
You can install the nightly pytorch build here: https://pytorch.org/get-started/locally/
88+
89+
```sh
90+
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129
91+
```
92+
7993
### Frontend (same for both)
8094
- Run Frontend:
8195
```

0 commit comments

Comments
 (0)