search

amd / InfinityHub-CI

MIT License
15 stars 5 forks source link

Updating TensorFlow and PyTorch containers #6

Closed dipietrantonio closed 1 year ago

dipietrantonio commented 1 year ago

Hi All, I have noticed that new TensorFlow + rocm containers are being published here https://hub.docker.com/r/rocm/tensorflow/tags instead of https://hub.docker.com/r/amdih/tensorflow . In addition, the containers published in amdih for Tensorflow and PyTorch are quite old, with PyTorch not being published under rocm like TensorFlow.

When can we expect a newer PyTorch container? And what is the default Docker Repo we should follow?

Thanks

dabolding commented 1 year ago

@dipietrantonio I am looking into this now for you. I hope to have information that I can share in the next couple days

dipietrantonio commented 1 year ago

Hi @dabolding, any news?

dabolding commented 1 year ago

Apologies for the delay. I believe these containers have been updated

dipietrantonio commented 1 year ago

I see now that the InfinityHub page now points to the DockerHub rocm organisation instead of amdih. So that is solved.