This repository was archived by the owner on Sep 19, 2022. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Expand file tree Collapse file tree 3 files changed +6
-6
lines changed Original file line number Diff line number Diff line change 1414 spec :
1515 containers :
1616 - name : pytorch
17- image : gcr.io/<your_project>/pytorch_dist_mnist :latest
17+ image : gcr.io/<your_project>/pytorch-dist-mnist :latest
1818 args : ["--backend", "gloo"]
1919 # Comment out the below resources to use the CPU.
2020 resources :
3030 spec :
3131 containers :
3232 - name : pytorch
33- image : gcr.io/<your_project>/pytorch_dist_mnist :latest
33+ image : gcr.io/<your_project>/pytorch-dist-mnist :latest
3434 args : ["--backend", "gloo"]
3535 # Comment out the below resources to use the CPU.
3636 resources :
Original file line number Diff line number Diff line change 1414 spec :
1515 containers :
1616 - name : pytorch
17- image : gcr.io/<your_project>/pytorch_dist_mnist :mpi
17+ image : gcr.io/<your_project>/pytorch-dist-mnist :mpi
1818 args : ["--backend", "mpi"]
1919 # Comment out the below resources to use the CPU.
2020 resources :
3030 spec :
3131 containers :
3232 - name : pytorch
33- image : gcr.io/<your_project>/pytorch_dist_mnist :mpi
33+ image : gcr.io/<your_project>/pytorch-dist-mnist :mpi
3434 args : ["--backend", "mpi"]
3535 # Comment out the below resources to use the CPU.
3636 resources :
Original file line number Diff line number Diff line change 1414 spec :
1515 containers :
1616 - name : pytorch
17- image : gcr.io/<your_project>/pytorch_dist_mnist :latest
17+ image : gcr.io/<your_project>/pytorch-dist-mnist :latest
1818 args : ["--backend", "nccl"]
1919 resources :
2020 limits :
2929 spec :
3030 containers :
3131 - name : pytorch
32- image : gcr.io/<your_project>/pytorch_dist_mnist :latest
32+ image : gcr.io/<your_project>/pytorch-dist-mnist :latest
3333 args : ["--backend", "nccl"]
3434 resources :
3535 limits :
You can’t perform that action at this time.
0 commit comments