gcc
、python3
、pip
等)。pip install torch --index-url https://download.pytorch.org/whl/cu113
)。conda install pytorch cudatoolkit=版本號
安裝,自動管理依賴。import torch
print(torch.__version__)
print(torch.cuda.is_available())
torch.nn.DataParallel
或DistributedDataParallel
擴展。關鍵提示:安裝前務必查閱PyTorch官方文檔,確認版本兼容性。