diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-01 19:43:09 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-08-01 19:43:09 -0700 |
| commit | 3479d02a769c1cbd6679caaf5695f382b558e36b (patch) | |
| tree | fcaa7aa4e7a11d3fa2e61390acacdd6627fa0e31 /tests | |
| parent | 8bf3e9faab6dfb04d676a5ea413530cdee09744c (diff) | |
Added some more docs and comments.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_cuda_setup_evaluator.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/test_cuda_setup_evaluator.py b/tests/test_cuda_setup_evaluator.py index 5da190d..119e21a 100644 --- a/tests/test_cuda_setup_evaluator.py +++ b/tests/test_cuda_setup_evaluator.py @@ -92,6 +92,9 @@ def test_get_cuda_runtime_lib_path__non_existent_dir(capsys, tmp_path): def test_full_system(): ## this only tests the cuda version and not compute capability + + # if CONDA_PREFIX exists, it has priority before all other env variables + # but it does not contain the library directly, so we need to look at the a sub-folder version = '' if 'CONDA_PREFIX' in os.environ: ls_output, err = bnb.utils.execute_and_return(f'ls -l {os.environ["CONDA_PREFIX"]}/lib/libcudart.so') |