#1281 numba mvc support

Kaggle · Aug 9, 2023 · ba33535 · ba33535
1 parent 2a8c4dd
commit ba33535
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -261,6 +261,10 @@ RUN pip install scipy \
     apt-get install -y pandoc && \
     pip install essentia
 
+# #1281 Install numba MVC support:
+pip install ptxcompiler-cu11 cubinlinker-cu11 --extra-index-url=https://pypi.nvidia.com
+ENV NUMBA_CUDA_ENABLE_MINOR_VERSION_COMPATIBILITY=1
+
 RUN apt-get install -y git-lfs && \
     /tmp/clean-layer.sh