yabs.io

Yet Another Bookmarks Service

Search

Results

Script for testing PyTorch support with AMD GPUs using ROCM

[https://gist.github.com/damico/484f7b0a148a0c5f707054cf9c0a0533] - 2024-03-08 01:01:10 - public:mzimmerm

amd, gpu, pytorch, rocm, test - 5 | id:1489851 -

Test for PyTorch and ROCm after installing ROCm.

Follow Tags

test - Please Log In To follow this tag
amd - Please Log In To follow this tag
gpu - Please Log In To follow this tag

Export: