pin nvidia-ml-py for breaking change