Dockerfile fix
This commit is contained in:
@@ -1,7 +1,9 @@
|
|||||||
## GPU-ready base image with CUDA 12 + cuDNN 9 runtime
|
## 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.
|
# 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.
|
# 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
|
# Keep python fast/quiet and pip lean
|
||||||
ENV PYTHONDONTWRITEBYTECODE=1 \
|
ENV PYTHONDONTWRITEBYTECODE=1 \
|
||||||
|
Reference in New Issue
Block a user