1. 程式人生 > 其它 >Pytorch:簡介和安裝

Pytorch:簡介和安裝

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 相關資源