install pytorch cpu code example

Example 1: get pytorch version

import torch
print(torch.__version__)

Example 2: mac install pytorch 3.6

conda install pytorch torchvision -c pytorch

Example 3: install pytorch cuda 10

# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch

Example 4: check cuda version pytorch

torch.version.cuda