-
Notifications
You must be signed in to change notification settings - Fork 91
Adding the Pytorch install options #134
base: master
Are you sure you want to change the base?
Conversation
|
@amathews-amd can you please review this. |
| sudo apt-get update | ||
| sudo apt-get upgrade | ||
|
|
||
| Recommended: Install using published PyTorch ROCm docker image: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Note: Currently, ROCm install version 3.3 is required ?
| sudo apt-get update | ||
| sudo apt-get upgrade | ||
|
|
||
| Recommended: Install using published PyTorch ROCm docker image: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Change Recommended: Install using published PyTorch ROCm docker image: to
Option 1 (Recommended): Install using published PyTorch ROCm docker image:
Install or update rocm-dev on the host system: should be it's own thing since its required for all 3 options.
| This step is optional but most PyTorch scripts will use torchvision to load models. E.g., running the pytorch examples requires torchvision. | ||
|
|
||
|
|
||
| Option 2: Install using PyTorch upstream docker file |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Option 2 title is missing
|
Keep same formatting in new additions as previous doc. |
https://ontrack-internal.amd.com/browse/SWDEV-303884