Skip to content

Commit c195472

Browse files
authored
Only enable numba mvc on GPU image
1 parent 3bb531e commit c195472

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Dockerfile.tmpl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -254,9 +254,11 @@ RUN pip install scipy \
254254
apt-get install -y pandoc && \
255255
pip install essentia
256256

257+
{{ if eq .Accelerator "gpu" }}
257258
# #1281 Install numba MVC support:
258259
RUN pip install ptxcompiler-cu11 cubinlinker-cu11 --extra-index-url=https://pypi.nvidia.com
259260
ENV NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1
261+
{{ end }}
260262

261263
RUN apt-get install -y git-lfs && \
262264
/tmp/clean-layer.sh

0 commit comments

Comments
 (0)