pytorch 1.6 code example

Example: torch conda

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch