What is needed to implement CUDA?

ghliu / pytorch-ddpg

Implementation of the Deep Deterministic Policy Gradient (DDPG) using PyTorch

Apache License 2.0

569 stars 157 forks source link

What is needed to implement CUDA? #9

Closed relh closed 5 years ago

relh commented 5 years ago

What is needed to implement GPUs?

relh commented 5 years ago

Okay, well I was able to make some small modifications and train successfully on my GPU without those things. I'll put up a PR to show what I did.

EDIT: I changed a lot and made the whole thing run in a multiprocess/distributed fashion and it's a lot to disentangle it. Gonna bail on the PR but ping me if you find this on Google and need help.

relh commented 5 years ago

move fast and break things? ;)

EDIT: Thanks for the help though!