use NCCL 2.11 in CUDA 11.4 tests
@@ -2,7 +2,7 @@ FROM nvidia/cuda:11.4.0-devel-ubuntu18.04
RUN export DEBIAN_FRONTEND=noninteractive && \
apt-get -qqy update && \
- apt-get -qqy install libnccl-dev=2.10.*+cuda11.4 libcudnn8-dev=8.2.*+cuda11.4 libcutensor-dev=1.3.* libcusparselt-dev=0.1.* && \
+ apt-get -qqy install libnccl-dev=2.11.*+cuda11.4 libcudnn8-dev=8.2.*+cuda11.4 libcutensor-dev=1.3.* libcusparselt-dev=0.1.* && \
apt-get -qqy install ccache
ENV PATH="/usr/lib/ccache:${PATH}"
