conda pytorch 1.5.1 code example

Example 1: torch conda

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

Example 2: how to install pytorch 0.4.1

pip install --trusted-host pypi.org --trusted-host files.pythonhosted.org torch===0.4.1 torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html