Skip to content

An app to check if a system has GPU(s) available

License

Notifications You must be signed in to change notification settings

FNNDSC/pl-gpu_test

Repository files navigation

pl-gpu-test

https://travis-ci.org/FNNDSC/gpu_test.svg?branch=master

See also: https://github.com/FNNDSC/dbg-nvidia-smi

An app to check the available GPUs

python gpu_test.py                                           \
    [-v <level>] [--verbosity <level>]                          \
    [--version]                                                 \
    [--man]                                                     \
    [--meta]                                                    \
    <inputDir>
    <outputDir>

gpu_test.py is a ChRIS-based application that...

[-v <level>] [--verbosity <level>]
Verbosity level for app. Not used currently.

[--version]
If specified, print version number.

[--man]
If specified, print (this) man page.

[--meta]
If specified, print plugin meta data.

This plugin can be run in two modes: natively as a python package or as a containerized docker image.

To run from PyPI, simply do a

pip install gpu_test

and run with

gpu_test.py --man /tmp /tmp

to get inline help. The app should also understand being called with only two positional arguments

gpu_test.py /some/input/directory /destination/directory

To run using docker, be sure to assign an "input" directory to /incoming and an output directory to /outgoing. Make sure that the $(pwd)/out directory is world writable!

Now, prefix all calls with

docker run --rm -v $(pwd)/out:/outgoing                             \
        fnndsc/pl-gpu_test gpu_test.py                        \

Thus, getting inline help is:

mkdir in out && chmod 777 out
docker run --rm -v $(pwd)/in:/incoming -v $(pwd)/out:/outgoing      \
        fnndsc/pl-gpu_test gpu_test.py                        \
        --man                                                       \
        /incoming /outgoing