pytorch 1.0

Screenshot from 2019-01-23 21-53-51.png727×733 124 KB

relevant source code: https://github.com/heronsystems/adeptRL/blob/master/adept/environments/managers/subproc_env_manager.py#L148-L193 3
even if I set os.environ[“CUDA_VISIBLE_DEVICES”] = “”, torch.cuda.is_available() returns False, but memory is still being allocated on GPU? maybe something to do with shared memory?