This repository was archived by the owner on Aug 28, 2025. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +18
-5
lines changed
lightning_examples/reinforce-learning-DQN Expand file tree Collapse file tree 3 files changed +18
-5
lines changed Original file line number Diff line number Diff line change 2323 # base ML image: mcr.microsoft.com/azureml/openmpi3.1.2-cuda10.2-cudnn8-ubuntu18.04
2424 image : " pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.10"
2525 # image: "pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime"
26- options : " -it --rm -- runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g"
26+ options : " -- runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g --name cd-container -v /usr/bin/docker:/tmp/docker:ro "
2727
2828 variables :
2929 ACCELERATOR : CPU,GPU
4040 python --version
4141 displayName: 'Image info & NVIDIA'
4242
43+ - script : |
44+ /tmp/docker exec -t -u 0 cd-container \
45+ sh -c "apt-get update && DEBIAN_FRONTEND=noninteractive apt-get -o Dpkg::Options::="--force-confold" -y install sudo"
46+ displayName: 'Install Sudo in container (thanks Microsoft!)'
47+
4348 - bash : |
4449 git fetch --all
4550 echo $(PUB_BRANCH)
5560 displayName: 'Set Git user'
5661
5762 - bash : |
63+ set -e
5864 pip --version
5965 pip install --requirement requirements.txt
6066 pip install --requirement requirements/data.txt
67+ sudo pip uninstall -y horovod
68+ sudo pip uninstall -y torchtext # FIXME: this is hotfix for https://github.com/PyTorchLightning/pytorch-lightning/pull/12676
6169 pip list
6270 displayName: 'Install dependencies'
6371
Original file line number Diff line number Diff line change 1919 # base ML image: mcr.microsoft.com/azureml/openmpi3.1.2-cuda10.2-cudnn8-ubuntu18.04
2020 image : " pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.10"
2121 # image: "pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime"
22- options : " -it --rm -- runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g"
22+ options : " -- runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g --name ci-container -v /usr/bin/docker:/tmp/docker:ro "
2323
2424 variables :
2525 ACCELERATOR : CPU,GPU
@@ -34,10 +34,18 @@ jobs:
3434 python --version
3535 displayName: 'Image info & NVIDIA'
3636
37+ - script : |
38+ /tmp/docker exec -t -u 0 ci-container \
39+ sh -c "apt-get update && DEBIAN_FRONTEND=noninteractive apt-get -o Dpkg::Options::="--force-confold" -y install sudo"
40+ displayName: 'Install Sudo in container (thanks Microsoft!)'
41+
3742 - bash : |
43+ set -e
3844 pip --version
3945 pip install --requirement requirements.txt
4046 pip install --requirement requirements/data.txt
47+ sudo pip uninstall -y horovod
48+ sudo pip uninstall -y torchtext # FIXME: this is hotfix for https://github.com/PyTorchLightning/pytorch-lightning/pull/12676
4149 pip list
4250 displayName: 'Install dependencies'
4351
Original file line number Diff line number Diff line change @@ -13,9 +13,6 @@ description: |
1313 2. Handle unsupervised learning by using an IterableDataset where the dataset itself is constantly updated during training
1414 3. Each training step carries has the agent taking an action in the environment and storing the experience in the IterableDataset
1515requirements :
16- - torchvision<=0.10
17- - torchaudio<=0.10
18- - torchtext<=0.10
1916 - gym
2017 - pygame
2118accelerator :
You can’t perform that action at this time.
0 commit comments