module 'torch.cuda' has no attribute 'Nimble'

[AmirFB]

I have installed nimble using the instruction but when I'm trying to run examples (run_inference.py and run_training.py) I get the following error.

Traceback (most recent call last): File "run_training.py", line 71, in <module> main(args) File "run_training.py", line 45, in main with closing(get_training_wrapper(model, dummy_input, dummy_label, args.mode, args.use_optimizer)) as training_wrapper: File "/home/amir/reps/nimble/experiment/utils.py", line 112, in get_training_wrapper wrapper = NimbleTrainingWrapper(model, dummy_input, dummy_label, criterion, optimizer, use_multi_stream=False) File "/home/amir/reps/nimble/experiment/utils.py", line 339, in __init__ nimble_model = torch.cuda.Nimble(model) AttributeError: module 'torch.cuda' has no attribute 'Nimble'

It seems like I have installed the original PyTorch and does not have the 'Nimble' class, while I have installed Nimble from the source using the instructions in the install.sh file.