When installing PyTorch using pip install torch
I get the following:
from tools.nnwrap import generate_wrappers as generate_nn_wrappers
ModuleNotFoundError: No module named 'tools.nnwrap'
I'm on macOS 10.14.4, Python 3.8.
I read through the thread here but it didn't help.
UPDATE 13/11/19
The accepted answer works.
As a side note, I finally setup my local environment using Anaconda, despite the meaty installation creating an enviroment has been easy:
#create the env
conda create -n mytensorflowenv python=3.7
#activate it
conda activate mytensorflowenv
then run the PyTorch installation for Conda:
conda install pytorch torchvision -c pytorch
All smooth now.