xuhuisheng / rocm-gfx803

185 stars 9 forks source link

Torch wheel and libmpi_cxx.so.40 #4

Closed Lunatik00 closed 1 year ago

Lunatik00 commented 2 years ago

The Dockerfile of rocm 4.5 full (rocm/dev-ubuntu-20.04:4.5-complete) works with the tensorflow wheel for python 3.8, but i have been unable to make it work with torch

image

I have searched but the solutions i get are mostly out of date and i have not been able to solve the issue by myself

xuhuisheng commented 2 years ago

You can try this:

sudo apt install -y liblmdb-dev libopencv-highgui-dev libopencv-contrib-dev libopenblas-dev
Lunatik00 commented 2 years ago

solved in #7 it should be added to the instructions

xuhuisheng commented 1 year ago

This issue didn't update one years ago. I will close it, please re-open if needed.