Skip to content

Commit

Permalink
specify curator dir
Browse files Browse the repository at this point in the history
Signed-off-by: Sarah Yurick <sarahyurick@gmail.com>
  • Loading branch information
sarahyurick committed Oct 3, 2024
1 parent 74a556d commit cf8bfea
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 3 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/gpuci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ jobs:
- name: Run PyTests with GPU mark
run: |
docker exec nemo-curator-container pytest -m gpu
docker exec nemo-curator-container pytest -m gpu --rootdir /opt/NeMo-Curator /opt/NeMo-Curator/tests
- name: Cleanup
run: |
Expand Down
2 changes: 0 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -13,5 +13,3 @@ git checkout $CURATOR_COMMIT
pip install cython pytest setuptools pip --upgrade
pip install --extra-index-url https://pypi.nvidia.com ".[cuda12x]"
EOF

ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12/compat/lib.real/

0 comments on commit cf8bfea

Please sign in to comment.