Update PyTorch script

tools/pytorch

@@ -44,7 +44,7 @@ sync() {
 install() {
     echo "pytorch install"
     venv
-    ${python} -m pip install --quiet --upgrade wheel
+    ${python} -m pip install --quiet --upgrade wheel psutil
     ${python} -m pip install --quiet --upgrade torch torchvision torchaudio torchao --pre --index-url https://download.pytorch.org/whl/nightly/cpu
     # ${python} -m pip install --quiet --upgrade torch-neuron --index-url https://pip.repos.neuron.amazonaws.com
     deactivate