Use rocm/pytorch:latest (#2613)

7e2103f8 · Jithun Nair · GitHub · 2e596e67 · 7e2103f8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

docker/Dockerfile.rocm docker/Dockerfile.rocm +1 -1

未找到文件。
--- a/docker/Dockerfile.rocm
+++ b/docker/Dockerfile.rocm
-FROM rocm/pytorch:rocm5.0.1_ubuntu18.04_py3.7_pytorch_1.10.0
+FROM rocm/pytorch:latest

 # install latest released version of deepspeed
 RUN pip install deepspeed && \