how to import pytorch in python code example Example: python pytorch conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch