Commit 581c3c26 authored by Felix Abecassis US's avatar Felix Abecassis US

Add nvtx package for CUDA 9.2

parent fd842331
Pipeline #23272335 passed with stages
in 35 minutes and 48 seconds
......@@ -6,5 +6,6 @@ ENV NCCL_VERSION 2.2.12
RUN apt-get update && apt-get install -y --no-install-recommends \
cuda-libraries-$CUDA_PKG_VERSION \
cuda-nvtx-$CUDA_PKG_VERSION \
libnccl2=$NCCL_VERSION-1+cuda9.2 && \
rm -rf /var/lib/apt/lists/*
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment