A method and an apparatus for selecting beams for non-codebook based uplink multiple-input and multiple-output are disclosed. In an embodiment, a method implemented by a user equipment (UE) includes receiving an allocation of at least one sounding reference signal (SRS) resource in an uplink subframe from a transmission and reception point (TRP) for a first set of uplink transmission beams, transmitting precoded SRSs in the at least one SRS resource to the TRP, wherein the precoded SRSs are precoded with UE-selected precoders and at least one rank and receiving an identification of two or more precoders for a second set of uplink transmission beams, wherein the second set of uplink transmission beams is selected from the first set of uplink transmission beams, and wherein the two or more precoders are based on the precoded SRSs.