One tensor is still on the CPU.
Could you post the line of code which throws this error?
Also, you should upgrade to 0.4.0.
You can find the install instructions on the website 2 and the migration guide here.