In case your 2D classification job runs on the master host (instead of an “external” worker host), modification of the master config.sh may be needed in v5.0.1 (see cuMemHostAlloc results in CUDA_ERROR_INVALID_VALUE (version 5.0.1) - #2 by wtempel)