Add CUDA_VISIBLE_DEVICES environment variable when using the --gpus flag #618
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: CI | |
| on: | |
| push: | |
| paths-ignore: | |
| - 'docs/**' | |
| pull_request: | |
| paths-ignore: | |
| - 'docs/**' | |
| jobs: | |
| ci: | |
| runs-on: ubuntu-20.04 | |
| steps: | |
| - uses: actions/checkout@v2 | |
| - name: Set up Python | |
| uses: actions/setup-python@v2 | |
| with: | |
| python-version: 3.8.5 | |
| - name: Install Python dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install setuptools wheel | |
| pip install -r ./requirements-test.txt | |
| - name: Install MLCube | |
| run: | | |
| cd mlcube | |
| python setup.py bdist_wheel | |
| pip install dist/mlcube-*.whl | |
| - name: Test MLCube CLI | |
| run: | | |
| cd mlcube | |
| pytest | |
| - name: Install MLCube Docker Runner | |
| run: | | |
| cd runners/mlcube_docker | |
| python setup.py bdist_wheel | |
| pip install dist/mlcube_docker-*.whl | |
| - name: Test MLCube Docker CLI | |
| run: | | |
| cd runners/mlcube_docker | |
| pytest | |
| - name: Install MLCube Singularity Runner | |
| run: | | |
| cd runners/mlcube_singularity | |
| python setup.py bdist_wheel | |
| pip install dist/mlcube_singularity-*.whl | |
| - name: Test MLCube Singularity CLI | |
| run: | | |
| cd runners/mlcube_singularity | |
| pytest | |
| - name: Install MLCube SSH Runner | |
| run: | | |
| cd runners/mlcube_ssh | |
| python setup.py bdist_wheel | |
| pip install dist/mlcube_ssh-*.whl | |
| - name: Test MLCube SSH CLI | |
| run: | | |
| cd runners/mlcube_ssh | |
| pytest | |
| - name: Install MLCube Kubernetes Runner | |
| run: | | |
| cd runners/mlcube_k8s | |
| python setup.py bdist_wheel | |
| pip install dist/mlcube_k8s-*.whl | |
| - name: Test MLCube Kubernetes CLI | |
| run: | | |
| cd runners/mlcube_k8s | |
| pytest | |
| - name: Install MLCube GCP Runner | |
| run: | | |
| cd runners/mlcube_gcp | |
| python setup.py bdist_wheel | |
| pip install dist/mlcube_gcp-*.whl | |
| - name: Test MLCube GCP CLI | |
| run: | | |
| cd runners/mlcube_gcp | |
| pytest | |
| - name: Install MLCube Kubeflow Runner | |
| run: | | |
| cd runners/mlcube_kubeflow | |
| python setup.py bdist_wheel | |
| pip install dist/mlcube_kubeflow-*.whl | |
| - name: Test MLCube Kubeflow CLI | |
| run: | | |
| cd runners/mlcube_kubeflow | |
| pytest |