search

Firstbober / rocm-pytorch-gfx803-docker

A Docker image based on rocm/pytorch with support for gfx803(Polaris 20-21 (XT/PRO/XL); RX580; RX570; RX560) and Python 3.8
MIT License
22 stars 2 forks source link

Add and pass self-test #2

Closed interfect closed 2 years ago

interfect commented 2 years ago

This adds a step to the end of the build where we actually import the installed torch module to make sure it loads cleanly.

I found that it doesn't, unless you create some symlinks to let it find the roctracer libraries at the .so versions that it is expecting.

I haven't observed any runtime problems as a result of just using the installed version (which I think is really a .4) as the .1 version that is expected, but they could be lurking. Maybe the roctracer I pinned before isn't precisely the right one?

Firstbober commented 2 years ago

I've used roctracer from around October 16th, but if symlink to newer works, then it's fine.