tests: Add cuSparse to cu110 image

flatironinstitute · Aug 22, 2023 · a397c61 · a397c61
1 parent 2dd7855
commit a397c61
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/cufinufft/docker/cuda11.0/Dockerfile-x86_64 b/tools/cufinufft/docker/cuda11.0/Dockerfile-x86_64
@@ -8,6 +8,7 @@ ENV CUDART_VERSION 11.0.171
 ENV CUFFT_VERSION 10.2.1.245
 ENV CURAND_VERSION 10.2.1.245
 ENV CUBLAS_VERSION 11.0.0.191
+ENV CUSPARSE_VERSION 11.1.1.245
 ENV NVPROF_VERSION 11.0.221
 ENV NVTX_VERSION 11.0.167
 ENV NVCC_VERSION 11.0.221
@@ -47,6 +48,7 @@ RUN yum install -y \
         libcufft-${CUDA_DASH_VERSION}-${CUFFT_VERSION}-1 \
         libcurand-${CUDA_DASH_VERSION}-${CURAND_VERSION}-1 \
         libcublas-${CUDA_DASH_VERSION}-${CUBLAS_VERSION}-1 \
+        libcusparse-${CUDA_DASH_VERSION}-${CUSPARSE_VERSION}-1 \
         cuda-nvtx-${CUDA_DASH_VERSION}-${NVTX_VERSION}-1 && \
     rm -rf /var/cache/yum/*
 
@@ -56,6 +58,7 @@ RUN yum install -y \
         libcufft-devel-${CUDA_DASH_VERSION}-${CUFFT_VERSION}-1 \
         libcurand-devel-${CUDA_DASH_VERSION}-${CURAND_VERSION}-1 \
         libcublas-devel-${CUDA_DASH_VERSION}-${CUBLAS_VERSION}-1 \
+        libcusparse-devel-${CUDA_DASH_VERSION}-${CUSPARSE_VERSION}-1 \
         cuda-nvprof-${CUDA_DASH_VERSION}-${NVPROF_VERSION}-1 \
         cuda-nvcc-${CUDA_DASH_VERSION}-${NVCC_VERSION}-1 && \
     rm -rf /var/cache/yum/*