Using PyTorch Hub Models#
This section shows you how to run pre-trained PyTorch models from the Torch Hub.
PyTorch models will be stored in:
$USER/.cache/huggingface
%USERPROFILE%\.cache\huggingface\hub
Note
In Docker, the models will be stored in /ROCM_APP/models/torch, this ensures that the model will be only downloaded once, even after stopping the Docker container.
Examples#
Make sure the environment is setup.
This guide provides a number of examples you can run out of the box. Navigate to get-started/pytorch/ and lunch jupyter lab. Some of them can only run if you have an AMD GPU in your system.
More Pre-trained Models#
If you would like to run a different model, PyTorch Hub hosts several models. Explore what PyTorch Hub has to offer https://pytorch.org/hub/.
Copyright (C) 2025 Advanced Micro Devices, Inc. All rights reserved.
SPDX-License-Identifier: MIT