Merge pull request #7 from fidelity/fix

fix Dockerfile MPI
fidelity · Jul 21, 2021 · 3ced64d · 3ced64d
2 parents 9806106 + e5b8ca5
commit 3ced64d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/stoke-gpu.Dockerfile b/docker/stoke-gpu.Dockerfile
@@ -110,4 +110,4 @@ WORKDIR /usr/src/code/apex
 RUN pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
 
 # Install stoke w/ MPI support
-RUN pip install --no-cache-dir --trusted-host pypi.python.org stoke[mpi]
+RUN pip install --no-cache-dir --trusted-host pypi.python.org stoke