كتاب الجنائز
11  
The Book of Prayer - Funerals
(9)
Chapter: The deceased is tormented because of his family's crying for him
(9)
باب الْمَيِّتُ يُعَذَّبُ بِبُكَاءِ أَهْلِهِ عَلَيْهِ ‏
Sahih Muslim 927 c

The same hadith is narrated on the authority of 'Umar through another chain of transmitters.

وَحَدَّثَنَاهُ مُحَمَّدُ بْنُ الْمُثَنَّى، حَدَّثَنَا ابْنُ أَبِي عَدِيٍّ، عَنْ سَعِيدٍ، عَنْ قَتَادَةَ، عَنْ سَعِيدِ، بْنِ الْمُسَيَّبِ عَنِ ابْنِ عُمَرَ، عَنْ عُمَرَ، عَنِ النَّبِيِّ صلى الله عليه وسلم قَالَ ‏ "‏ الْمَيِّتُ يُعَذَّبُ فِي قَبْرِهِ بِمَا نِيحَ عَلَيْهِ ‏"‏ ‏.‏
Reference : Sahih Muslim 927c
In-book reference : Book 11, Hadith 22
USC-MSA web (English) reference : Book 4, Hadith 2017
  (deprecated numbering scheme)