yabs.io

Yet Another Bookmarks Service

Search

Results

Install PyTorch for ROCm — Use ROCm on Radeon GPUs

[https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/install-pytorch.html] - 2024-03-08 01:45:24 - public:mzimmerm

amd, gpu, install, post, pytorch, rocm, validate - 7 | id:1489852 -

Install PyTorch corresponding to rocm versions from Pytorch prebuild by AMD. Also has post-install validation

Follow Tags

post - Please Log In To follow this tag
gpu - Please Log In To follow this tag
validate - Please Log In To follow this tag

Export: