|
@@ -36,6 +36,7 @@ WORKDIR /workspace
|
|
|
# install build and runtime dependencies
|
|
|
COPY requirements-common.txt requirements-common.txt
|
|
|
COPY requirements-cuda.txt requirements-cuda.txt
|
|
|
+RUN pip install packaging wheel
|
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
|
python3 -m pip install -r requirements-cuda.txt
|
|
|
|