Skip to content

Commit

Permalink
Switch base image to nvidia/cuda for GPU support
Browse files Browse the repository at this point in the history 
Updated Dockerfile to use nvidia/cuda:12.3.2-base-ubuntu22.04 as the base image to enable GPU support. This change is made for better compatibility with CUDA 12 and GPU acceleration, aligning with the requirement to run applications that make extensive use of GPU resources. The previous image from chainguard was replaced to exploit the latest CUDA version's features and performance improvements. The comment on the ubuntu:22.04 line preserves the option for a future switch if needed.

Refs open-webui#1234 (if there's an associated issue)
  • Loading branch information
@yousecjoe
yousecjoe committed Mar 15, 2024
1 parent c9bb592 commit f9e8d34
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,9 @@ RUN npm run build


#FROM python:3.11-slim-bookworm as base
FROM cgr.dev/chainguard/pytorch-cuda12 as base
#FROM cgr.dev/chainguard/pytorch-cuda12 as base
#FROM ubuntu:22.04 as base
FROM nvidia/cuda:12.3.2-base-ubuntu22.04 as base

ENV ENV=prod
ENV PORT ""
Expand Down

0 comments on commit f9e8d34

Please sign in to comment.