schrodinger.seam.testing.stu_tests.gpu_worker module

Verify that a Seam GPU worker runs properly on a configured remote gpu host.

Usage:

$SCHRODINGER/run python3 -m schrodinger.seam.testing.stu_tests.gpu_worker -GPUHOST <gpu-host>:1

<gpu-host> must be a configured remote GPU host, not ‘localhost’.

schrodinger.seam.testing.stu_tests.gpu_worker.get_job_spec_from_args(argv)
class schrodinger.seam.testing.stu_tests.gpu_worker.CheckGPU(*unused_args, **unused_kwargs)

Bases: _GetComputeHintsMixin, CheckGPU

schrodinger.seam.testing.stu_tests.gpu_worker.parse_args()
schrodinger.seam.testing.stu_tests.gpu_worker.main(args)