marine grease would be good, like for prop shafts or for trailer wheel bearings, if you plan on driving through alot of water, otherwise normal black chasiss or multi purpose would be best and is lower cost
normal yellow wheel bearing grease is for resisting up to 500 degrees, your driveshaft is not gonna get to 500 degrees,.......
i wouldnt use it personall,y but up to you....