Update for CUDA 5.5 and better installation using pip