add pytorch to conda environment code example

Example: pytorch conda environment

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch