- PyTorch深度學習與企業級項目實戰
- 宋立桓 宋立林
- 207字
- 2024-12-31 19:17:15
2.2 在Windows環境下安裝CPU版的PyTorch
PytTorch是基于Python開發的,首先需要安裝Python,Python的安裝很簡單,這里不再贅述。而Windows用戶能直接通過conda、pip和源碼編譯三種方式來安裝PyTorch。
打開PyTorch官網(https://pytorch.org/),在主頁中根據自己的計算機選擇Linux、Mac或Windows系統,如圖2-8所示,系統將給出對應的安裝命令語句,比如這里為pip3 install torch torchvision torchaudio。

圖2-8
通過安裝命令安裝PyTorch 2.1.0,結果如下:


驗證PyTorch是否安裝成功,執行如下命令,注意命令中的雙下畫線:
print(torch.__version__) print(torch.version.cuda) print(torch.cuda.is_available())
命令執行結果如下:

如果沒有報錯,則說明PyTorch安裝成功。