I’m not running out of memory. I’ve monitored the GPU mem allocated and it’s no more than 1.7GB. I’ve also disabled TDR and it didn’t change anything.
On top of that I’ve tried it on other PC’s with a couple different recent drivers

My other Windows 10 machine, GTX 960 (Driver 456.71 and 436.48), 48GB Corsair memory, Ryzen 1300
Friends Windows 10 machine, GTX 960(456.71), 16GB RAM, i5-3570k
Friends Windows 10 machine GTX 980TI(456.71), 32GB RAM, Ryzen 2700X

It DID work fine on my other machine when I tried it on my Ubuntu 16 boot.