Joseph Musielewicz

icon indicating a website link https://jmusiel.github.io/

icon company Carnegie Mellon University icon location Pittsburgh, PA icon location PhD student in the Ulissi Group

Results 2 comments of Joseph Musielewicz

pipeline_parallel - ModuleNotFoundError: No module named 'amp_C'

I have the same issue in pipeline_parallels, same stack trace and everything I am on the Python-only build of apex

pipeline_parallel - ModuleNotFoundError: No module named 'amp_C'

> I have the same issue. The workaround that worked for me is to use the version prior to the last commit here: https://github.com/NVIDIA/apex/tree/3303b3e7174383312a3468ef390060c26e640cb1 Yes this worked for me as...