Skip to content

Commit

Permalink
set opencl version
Browse files Browse the repository at this point in the history
  • Loading branch information
jonafeucht committed Nov 9, 2024
1 parent 58db94a commit 8e83c24
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
4 changes: 3 additions & 1 deletion Dockerfile.opencl.amd.rusticl
Original file line number Diff line number Diff line change
Expand Up @@ -78,8 +78,10 @@ RUN chmod 666 /dev/dri/card* || true
# Set environment variables for OpenCL
ENV OCL_ICD_VENDORS="/etc/OpenCL/vendors"
ENV OPENCL_VENDOR_PATH="/etc/OpenCL/vendors"

ENV RUSTICL_ENABLE=radeonsi
ENV RUSTICL_FEATURES="f16,f64"
ENV RUSTICL_CL_VERSION=2.1
ENV RUSTICL_FEATURES=fp16,fp64

COPY requirements.txt /app/requirements.txt
RUN /app/venv/bin/pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
Expand Down
6 changes: 4 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ services:
### **AMD GPU Support**
> [!WARNING]
> Unless AMD starts to officially support older Cards again (e.g. gfx8), ROCm will not be available.
> Unless AMD starts to officially support older cards again (e.g. gfx8), ROCm will not be available.
**OpenCL Rusticl (AMD64):**
> [!CAUTION]
Expand Down Expand Up @@ -119,6 +119,8 @@ services:
- "render"
```
**OpenCL Clover (AMD64):**
> [!CAUTION]
> Clover is only recommended for legacy cards (e.g. gfx8). If you're using gfx10 or newer, use Rusticl instead.
```yml
services:
image_video_classification_cuda:
Expand Down Expand Up @@ -176,7 +178,7 @@ Only tested with consumer grade hardware and only on Linux based systems.
- GTX 950
- RTX 3060 Ti

#### AMD GPU (OpenCL)
#### AMD GPU (OpenCL Clover & Rusticl)
- RX 570 4GB
- RX 580 8GB
- RX 6600 XT
Expand Down

0 comments on commit 8e83c24

Please sign in to comment.