added steps to run horovod on multiple containers

This commit is contained in:
Benjamin Jacob Reji
2021-05-26 16:14:45 +04:00
parent 31b9e7e934
commit 5a164749b3
2 changed files with 28 additions and 1 deletions

View File

@@ -10,7 +10,7 @@
# from consol/ubuntu-xfce-vnc
# due to dependency issues vnc is still work in progress
from nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04
from nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
# Switch to root user to install additional software
USER 0