pytorch @ documentation code example Example: python pytorch conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch