upda (huggingface#5)

rraminen · Apr 10, 2021 · f0c7a14 · f0c7a14
1 parent 5908372
commit f0c7a14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/rocm/huggingface_zero.Dockerfile b/docker/rocm/huggingface_zero.Dockerfile
@@ -1,5 +1,5 @@
 # Select base Image
-FROM jithunnair/pytorch:rocm4.1_ubuntu18.04_py3.6_pytorch_deepspeed
+FROM rocm/deepspeed:rocm4.1_ubuntu18.04_py3.6_pytorch
 
 # Install dependencies
 RUN apt update && apt install -y \