Skip to content

Commit

Permalink
Update config.yml (#2226)
Browse files Browse the repository at this point in the history
Temporarily installing pytorch v1.9.1 manually instead of using pre-built docker image (not yet available).
  • Loading branch information
vfdev-5 authored Sep 27, 2021
1 parent fc1e176 commit e48d8db
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -95,6 +95,10 @@ run_pytorch_container: &run_pytorch_container
docker run --gpus=all --rm -itd --shm-size 16G -v ${wd}:/ignite -w /ignite --name pthd << pipeline.parameters.pytorch_stable_image >>
docker exec -it pthd nvidia-smi
docker exec -it pthd ls
# temporarily manually install v1.9.1
# https://github.com/pytorch/ignite/pull/2211#issuecomment-927080841
export update_pth_cmd='conda install -y pytorch==1.9.1 -c pytorch -c nvidia'
docker exec -it pthd /bin/bash -c "$update_pth_cmd"
run_pytorch_devel_container: &run_pytorch_devel_container
- run:
Expand All @@ -105,6 +109,10 @@ run_pytorch_devel_container: &run_pytorch_devel_container
docker run --gpus=all --rm -itd --shm-size 16G -v ${wd}:/ignite -w /ignite --name pthd << pipeline.parameters.pytorch_stable_image_devel >>
docker exec -it pthd nvidia-smi
docker exec -it pthd ls
# temporarily manually install v1.9.1
# https://github.com/pytorch/ignite/pull/2211#issuecomment-927080841
export update_pth_cmd='conda install -y pytorch==1.9.1 -c pytorch -c nvidia'
docker exec -it pthd /bin/bash -c "$update_pth_cmd"
install_dependencies: &install_dependencies
- run:
Expand Down

0 comments on commit e48d8db

Please sign in to comment.