StackNN icon indicating copy to clipboard operation
StackNN copied to clipboard
Published 6 years ago • verified publisher icon viking-sudo-rm

CUDA compatibility

Open viking-sudo-rm opened this issue 6 years ago • 1 comments

I'm not quite sure if the StackNN code is fully CUDA compatible, but recently I had some issues trying to use it.

Should investigate this and add cuda options everywhere in structs/control_layer.

viking-sudo-rm avatar May 29 '19 00:05 viking-sudo-rm

https://pytorch.org/docs/stable/notes/cuda.html

viking-sudo-rm avatar May 30 '19 16:05 viking-sudo-rm