Why is my code not using the GPU on Alex?