Skip to content

Commit b3c2954

Browse files
authored
Merge pull request Kaggle#416 from Kaggle/cupy
Add cupy and pynvrtc to GPU Dockerfile for custom networks in CUDA
2 parents 694c413 + 02ffc7d commit b3c2954

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

gpu.Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,4 +54,6 @@ RUN pip uninstall -y tensorflow && \
5454

5555
# Install GPU-only packages
5656
RUN pip install pycuda && \
57+
pip install cupy-cuda92 && \
58+
pip install pynvrtc && \
5759
/tmp/clean-layer.sh

0 commit comments

Comments
 (0)