Azure ML environment

Ive set up an environment on an Azure ML VM but the code runs extremely slowly (much slower than my laptop), so Im wondering if all dependencies have been installed correctly. Is there a preferred YAML file available for 5.0.2 I can use/edit?
Any other pointers gratefully received!