File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -108,6 +108,7 @@ RUN <<EOT bash
108108 apt-get update && \
109109 apt-get install -y --no-install-recommends \
110110 cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
111+ cuda-nvrtc-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
111112 libcufft-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
112113 libcurand-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
113114 libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
Original file line number Diff line number Diff line change @@ -126,6 +126,7 @@ RUN <<EOT bash
126126 apt-get update && \
127127 apt-get install -y --no-install-recommends \
128128 cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
129+ cuda-nvrtc-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
129130 libcufft-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
130131 libcurand-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
131132 libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
You can’t perform that action at this time.
0 commit comments