jimmy60504's picture
Upload folder using huggingface_hub
255b5a3 verified
Raw
History Blame Contribute Delete
982 Bytes
# syntax=docker/dockerfile:1.7
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:26.03-py3
FROM ${BASE_IMAGE}
ENV DEBIAN_FRONTEND=noninteractive \
PIP_NO_CACHE_DIR=1 \
PYTHONDONTWRITEBYTECODE=1 \
PYTHONUNBUFFERED=1
WORKDIR /workspace
# The NVIDIA PyTorch image ships with a pip constraints file. We only drop the
# hard pin on packaging tools so adding the repo's Python stack stays practical.
RUN python - <<'PY'
from pathlib import Path
constraint_path = Path("/etc/pip/constraint.txt")
if constraint_path.exists():
lines = []
for line in constraint_path.read_text().splitlines():
if line.startswith(("pip==", "setuptools==", "wheel==")):
continue
lines.append(line)
constraint_path.write_text("\n".join(lines) + "\n")
PY
COPY docker/requirements-gx10.txt /tmp/requirements-gx10.txt
RUN python -m pip install -r /tmp/requirements-gx10.txt
COPY hello_qiskit.py qft_demo.py smoke_test.py README.md ./
CMD ["python", "smoke_test.py"]