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)¶