Dockerfile fix
This commit is contained in:
@@ -1,7 +1,9 @@
|
||||
## GPU-ready base image with CUDA 12 + cuDNN 9 runtime
|
||||
# If you don't have an NVIDIA GPU or the NVIDIA Container Toolkit, this image still runs on CPU.
|
||||
# For smaller CPU-only images, you can switch back to python:3.11-slim.
|
||||
FROM nvidia/cuda:12.4.1-cudnn9-runtime-ubuntu22.04
|
||||
# Use a widely available CUDA + cuDNN runtime tag. If you prefer newer CUDA,
|
||||
# adjust this to a tag that exists on Docker Hub.
|
||||
FROM nvidia/cuda:12.3.2-cudnn8-runtime-ubuntu22.04
|
||||
|
||||
# Keep python fast/quiet and pip lean
|
||||
ENV PYTHONDONTWRITEBYTECODE=1 \
|
||||
|
Reference in New Issue
Block a user