Skip to content

Commit e48d8db

Browse files
authored
Update config.yml (#2226)
Temporarily installing pytorch v1.9.1 manually instead of using pre-built docker image (not yet available).
1 parent fc1e176 commit e48d8db

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

.circleci/config.yml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,10 @@ run_pytorch_container: &run_pytorch_container
9595
docker run --gpus=all --rm -itd --shm-size 16G -v ${wd}:/ignite -w /ignite --name pthd << pipeline.parameters.pytorch_stable_image >>
9696
docker exec -it pthd nvidia-smi
9797
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"
98102
99103
run_pytorch_devel_container: &run_pytorch_devel_container
100104
- run:
@@ -105,6 +109,10 @@ run_pytorch_devel_container: &run_pytorch_devel_container
105109
docker run --gpus=all --rm -itd --shm-size 16G -v ${wd}:/ignite -w /ignite --name pthd << pipeline.parameters.pytorch_stable_image_devel >>
106110
docker exec -it pthd nvidia-smi
107111
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"
108116
109117
install_dependencies: &install_dependencies
110118
- run:

0 commit comments

Comments
 (0)