[GPU Image] Fix Cuda link for horovod (#1180)

mlrun · Aug 3, 2021 · f5efea1 · f5efea1
1 parent acf5e3d
commit f5efea1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dockerfiles/models-gpu/Dockerfile b/dockerfiles/models-gpu/Dockerfile
@@ -82,7 +82,7 @@ ENV OMPI_ALLOW_RUN_AS_ROOT_CONFIRM=1
 
 # TODO: MAKEFLAGS="-j1" work around some transient concurrency problem with installing horovod remove it when
     #   possible (should be safe to remove if it works ~5 times without it)
-RUN ldconfig /usr/local/cuda-10.1/targets/x86_64-linux/lib/stubs && \
+RUN ldconfig /usr/local/cuda-11.0/targets/x86_64-linux/lib/stubs && \
     MAKEFLAGS="-j1" HOROVOD_GPU_OPERATIONS=NCCL HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 \
         python -m pip install 'horovod~=0.20.0' && \
     ldconfig