Move model and data to GPU if available in basic tutorial

[guojing0]

Description

This PR improves the "Optimizing Model Parameters" by moving model and data to GPU if available.

Checklist

• The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
• Only one issue is addressed in this pull request
• Labels from the issue that this PR is fixing are added to this pull request
• No unnecessary issues are included into this pull request.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/3535

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• [ROCm][CI] Machines under the label linux.rocm.gpu.2, label linux.rocm.gpu.4, linux.rocm.gpu.gfx1100 are undergoing maintenance.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[AlannaBurke]

https://docs-preview.pytorch.org/pytorch/tutorials/3535/beginner/basics/optimization_tutorial.html

LGTM

[AlannaBurke]

https://docs-preview.pytorch.org/pytorch/tutorials/3536/intermediate/torchvision_tutorial.html

LGTM

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as stale.
Feel free to remove the stale label if you feel this was a mistake.
If you are unable to remove the stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
stale pull requests will automatically be closed after 30 days of inactivity.

[guojing0]

no-stable