how to install pytorch in python code example

Example 1: conda install pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

Example 2: pytorch install

pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

Example 3: install torch anaconda

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

Example 4: install pytorch

NOTE: Python 3.9 users will need to add '-c=conda-forge' for installationconda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch