File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -120,6 +120,7 @@ RUN <<EOT bash
120
120
apt-get update && \
121
121
apt-get install -y --no-install-recommends \
122
122
cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
123
+ libcufft-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
123
124
libcurand-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
124
125
libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
125
126
libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
@@ -139,6 +140,7 @@ RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \
139
140
apt-get update && \
140
141
apt-get install -y --no-install-recommends \
141
142
cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
143
+ libcufft-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
142
144
libcurand-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
143
145
libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
144
146
libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
You can’t perform that action at this time.
0 commit comments