A simplified two-dimensional axisymmetric model was established based on a typical continental sedimentary basin in China to simulate the thermal evolution of wellbore and reservoir during the injection of CO 2 by taking consideration of lithology heterogeneity of reservoir. By comparing with two simple one-dimensional theory models, the lithology heterogeneity influence on CO 2 mass flow rate distribution along depth in the wellbore is identified. Results suggested that the interaction of multiple layers in the heterogeneous reservoir will influence the CO 2 mass flow rate distribution along depth in the wellbore so as to impact the corresponding temperature and pressure evolution in the wellbore and reservoir. Layer burial depth (or relative location), porosity, permeability and thickness are all important factors that affect CO 2 mass flow rate in wellbore. The variation of CO 2 mass flow rate in the wellbore will change the CO 2 temperature flowing into each layers through impact the heat extraction from rocks, compressibility of CO 2 and potential energy loss, and by varying the CO 2 hydrostatic pressure and pressure drop due to friction to determine the CO 2 injection pressure. Layer burial depth, porosity, permeability and thickness are all important factors that affect the CO 2 mass flow rate distribution in the wellbore. This study may help deepen our understanding of CO 2 flow and thermal evolution in the actual heterogeneous reservoir and provide important knowledge supplement for the liquid injection (especially CO 2 ) into underground, such as deep saline aquifer, depleted oil/gas reservoir and coal bed.