Yes
The same way u always do.
Tho it’s preferable that u use the cpu environment on colab rather than the gpu environment except u are using colab pro (which is a paid version)
The gpu environment in colab can be really unstable sometimes, but u can always try it out.