Spaces:

dkdaniz
/

katara

Paused

dkdaniz commited on Oct 8, 2023

Commit

8b4dc3e

•

1 Parent(s): 7c5e1c1

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -1,8 +1,7 @@
 # syntax=docker/dockerfile:1
 # Build as `docker build . -t localgpt`, requires BuildKit.
 # Run as `docker run -it --mount src="$HOME/.cache",target=/root/.cache,type=bind --gpus=all localgpt`, requires Nvidia container toolkit.
-ARG CUDA_IMAGE="12.2.0-devel-ubuntu20.04"
-FROM nvidia/cuda:${CUDA_IMAGE}
 RUN apt-get update && apt-get upgrade -y \
     && apt-get install -y git build-essential \

 # syntax=docker/dockerfile:1
 # Build as `docker build . -t localgpt`, requires BuildKit.
 # Run as `docker run -it --mount src="$HOME/.cache",target=/root/.cache,type=bind --gpus=all localgpt`, requires Nvidia container toolkit.
+FROM nvidia/cuda:11.7.1-runtime-ubuntu22.04
 RUN apt-get update && apt-get upgrade -y \
     && apt-get install -y git build-essential \