pytorch gpu conda install code example Example: conda install pytorch conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch