File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -95,6 +95,10 @@ run_pytorch_container: &run_pytorch_container
95
95
docker run --gpus=all --rm -itd --shm-size 16G -v ${wd}:/ignite -w /ignite --name pthd << pipeline.parameters.pytorch_stable_image >>
96
96
docker exec -it pthd nvidia-smi
97
97
docker exec -it pthd ls
98
+ # temporarily manually install v1.9.1
99
+ # https://github.com/pytorch/ignite/pull/2211#issuecomment-927080841
100
+ export update_pth_cmd='conda install -y pytorch==1.9.1 -c pytorch -c nvidia'
101
+ docker exec -it pthd /bin/bash -c "$update_pth_cmd"
98
102
99
103
run_pytorch_devel_container : &run_pytorch_devel_container
100
104
- run :
@@ -105,6 +109,10 @@ run_pytorch_devel_container: &run_pytorch_devel_container
105
109
docker run --gpus=all --rm -itd --shm-size 16G -v ${wd}:/ignite -w /ignite --name pthd << pipeline.parameters.pytorch_stable_image_devel >>
106
110
docker exec -it pthd nvidia-smi
107
111
docker exec -it pthd ls
112
+ # temporarily manually install v1.9.1
113
+ # https://github.com/pytorch/ignite/pull/2211#issuecomment-927080841
114
+ export update_pth_cmd='conda install -y pytorch==1.9.1 -c pytorch -c nvidia'
115
+ docker exec -it pthd /bin/bash -c "$update_pth_cmd"
108
116
109
117
install_dependencies : &install_dependencies
110
118
- run :
You can’t perform that action at this time.
0 commit comments