Skip to content

Commit 8e15dd3

Browse files
authored
Merge pull request #1282 from Kaggle/numba-mvc
#1281 numba mvc support
2 parents a73246b + 5324ff3 commit 8e15dd3

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

Dockerfile.tmpl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,10 @@ RUN pip install scipy \
261261
apt-get install -y pandoc && \
262262
pip install essentia
263263

264+
# #1281 Install numba MVC support:
265+
RUN pip install ptxcompiler-cu11 cubinlinker-cu11 --extra-index-url=https://pypi.nvidia.com
266+
ENV NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1
267+
264268
RUN apt-get install -y git-lfs && \
265269
/tmp/clean-layer.sh
266270

0 commit comments

Comments
 (0)