install puytorch code example Example: torch conda conda install pytorch torchvision cudatoolkit=10.2 -c pytorch