So few things to check:

When installing with pip, did you take the wheel for cuda 9.0?
When installing from source, was your cuda install detected properly? If so is it detection the install that you want it to use?
Do you have the cuda samples? Do they run properly? Does nvidia-smi run properly?