Use default device manager in PyTorch - Githubissues

DeepX-inc / machina

Control section: Deep Reinforcement Learning framework

MIT License

278 stars 45 forks source link

Use default device manager in PyTorch #214

Open rarilurelo opened 5 years ago

rarilurelo commented 5 years ago

Using torch.cuda.set_device instead of machina.utils.set_device.