Installation

pip3 install torch torchvision