安裝PyTorch可以通過以下步驟完成:
python --version
pip install --upgrade pip
pip install torch==1.7.0+cu102 torchvision==0.8.1+cu102 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
import torch
print(torch.__version__)
如果沒有出現錯誤,并且成功打印出PyTorch的版本號,那么PyTorch已經成功安裝在你的系統中。