pytorch supported cude toolkit code example Example: mac install pytorch 3.6 conda install pytorch torchvision -c pytorch