Sure! If you are using a nn.Sequential model, you can just register the forward hook to the nn.ReLU module.
In my example just call
model[1].register_forward_hook(get_activation('layer0_relu'))`
...
plt.matshow(activations['layer0_relu'])