Pytorch:簡介和安裝
阿新 • • 發佈:2022-05-17
PyTorch 簡介
PyTorch 是一個開源的 Python 機器學習庫,提供兩個高階功能:1、具有強大的GPU加速的張量計算;2、包含自動求導系統的深度神經網路。
PyTorch 的優點:簡潔且高效快速、設計符合人類的思維、入門簡單、FAIR的支援保證可以持續更新、作者親自維護交流論壇。已相容 Windows(CUDA,CPU)、MacOS(CPU)、Linux(CUDA,ROCm,CPU)。
注:框架只是給我們提供了輪子(工具),讓我們造汽車(產品)更加方便,最重要的還是我們個人的科學素養的提升(設計)。
PyTorch 的安裝
登入 PyTorch官網 → 根據自己情況選擇命令並複製(推薦使用 Anaconda 配套)→ 執行命令
檢查是否安裝成功:輸入python → 輸入import torch,沒有報錯則安裝成功
注:Anaconda、Pycharm 和 Jupyter 是常用的組合,相關安裝教程可在他處尋找,本文不再贅述。
顯示卡檢查
使用工作管理員檢視自己是否有 NVIDIA 的獨立顯示卡及其型號。
NVIDIA GeForce GT 620 屬於入門級獨立顯示卡、低端顯示卡,較差。
BUG總結
輸入 Pytorch 的安裝命令後報錯,如下圖所示:
按照提示,手動刪除對應的檔案和資料夾;再輸入安裝pytorch的命令即可。
PyTorch 相關資源
-
Awesome-pytorch-list
- PyTorch官方文件:官方釋出的文件,十分豐富。
- Pytorch-handbook:GitHub上已經收穫14.8K,pytorch手中書。
- PyTorch官方社群:在這裡你可以和開發pytorch的人們進行交流
- 深入淺出 PyTorch:Datawhale 官方教程