HomeBlog其他如何在 Windows 10 / 11 上安裝 PyTorch [Nvidia AMD GPU 和 CPU]

如何在 Windows 10 / 11 上安裝 PyTorch [Nvidia AMD GPU 和 CPU]

cover_img
  1. 安裝 PyTorch 的先決條件
  2. 導航至 PyTorch 首頁
  3. 選擇合適的計算平台
  4. 檢查 CUDA 版本
  5. 下載 CUDA 工具包
  6. 安裝 CUDA
  7. 安裝 Visual Studio
  8. 完成 CUDA 安裝
  9. 完成 PyTorch 安裝
  10. 驗證 PyTorch 安裝
  11. 常見問題

安裝 PyTorch 的先決條件

在安裝 PyTorch 之前,請確保您的計算機上已安裝 Python 和 pip。如果您尚未安裝它們,請參考可靠的在線資源以獲取設置指南。當您準備好 Python 和 pip 後,便可以繼續安裝 PyTorch。

導航至 PyTorch 首頁

要開始安裝過程,請訪問官方 PyTorch 網站。向下滾動直到找到安裝選項。選擇穩定版本的 PyTorch,目前是版本 2.1.1。選擇您的操作系統;例如,如果您使用的是 Windows,請選擇該選項。由於您沒有安裝 Conda,因此將包留作 pip,並確保選擇 Python 作為語言。

選擇合適的計算平台

計算平台選項至關重要,因為它與您的顯示卡有關。如果您有 Nvidia 顯示卡,請選擇相應的 CUDA 版本。對於 AMD 顯示卡,選擇 ROCm。如果您的系統沒有 GPU,則選擇 CPU 選項。請注意,AMD 選項不適用於 Windows 用戶;它需要 Linux 操作系統。

檢查 CUDA 版本

在執行 pip install 命令之前,檢查系統上安裝的 CUDA 版本非常重要。打開 PowerShell 並輸入命令 'nvcc --version'。如果未安裝 CUDA,您需要從 Nvidia 網站下載它。

下載 CUDA 工具包

要安裝 CUDA,請在瀏覽器中搜索 'install CUDA 12.1' 並點擊 Nvidia 官方網站的第一個結果。選擇您的操作系統、架構(x86 64)和 Windows 版本。選擇 'exe local' 安裝程序類型,並確保在開始下載之前有至少 3.1 GB 的可用空間。

安裝 CUDA

下載完成後,導航到您的下載文件夾並打開可執行文件。您可能需要刷新頁面以查看下載。雙擊可執行文件並允許其通過用戶帳戶控制。選擇提取路徑並繼續提取。按照提示操作,同意許可協議並選擇快速安裝選項。

安裝 Visual Studio

在 CUDA 安裝過程中,您可能會遇到一條消息,指示需要支持的 Visual Studio 版本。為了解決此問題,請在瀏覽器中打開新標籤並搜索 Visual Studio。下載免費的社區版。下載完成後,打開安裝程序並選擇您希望安裝的包,確保包括 Python 開發包。

完成 CUDA 安裝

安裝 Visual Studio 後,返回 CUDA 安裝過程。再次打開 CUDA 12.1 可執行文件,選擇提取路徑,並繼續安裝。按照提示完成安裝過程。完成後,通過在 PowerShell 中運行 'nvcc --version' 來驗證安裝,以確認 CUDA 12.1 現在已安裝。

完成 PyTorch 安裝

CUDA 安裝成功後,您現在可以返回 PyTorch 安裝過程。複製 PyTorch 網站上提供的安裝命令並將其粘貼到 PowerShell 中。執行該命令以在您的 Windows 10 或 11 操作系統上安裝 PyTorch。

驗證 PyTorch 安裝

為了確保 PyTorch 已正確安裝,請在 PowerShell 中運行幾個命令。首先輸入 'python' 以進入 Python 環境。接下來,通過輸入 'import torch' 來導入 PyTorch。要檢查 CUDA 是否可用,使用命令 'torch.cuda.is_available()'。如果返回 true,恭喜您!您已成功安裝 PyTorch。

常見問題

問:安裝 PyTorch 的先決條件是什麼?
答:在安裝 PyTorch 之前,請確保您的計算機上已安裝 Python 和 pip。
問:我如何導航至 PyTorch 首頁以進行安裝?
答:訪問官方 PyTorch 網站並向下滾動以找到安裝選項。
問:我如何選擇合適的計算平台以安裝 PyTorch?
答:對於 Nvidia 顯示卡,選擇相應的 CUDA 版本,對於 AMD,選擇 ROCm,或者如果沒有 GPU,則選擇 CPU 選項。
問:我如何檢查系統上安裝的 CUDA 版本?
答:打開 PowerShell 並輸入命令 'nvcc --version'。
問:我如何下載 CUDA 工具包?
答:在瀏覽器中搜索 'install CUDA 12.1' 並選擇 Nvidia 官方網站的第一個結果。
問:安裝 CUDA 涉及哪些步驟?
答:打開下載的可執行文件,選擇提取路徑,並按照提示完成安裝。
問:如果在 CUDA 安裝過程中需要安裝 Visual Studio,我該怎麼辦?
答:下載 Visual Studio 的社區版,確保包括 Python 開發包。
問:我如何完成 CUDA 安裝?
答:重新打開 CUDA 12.1 可執行文件,選擇提取路徑,並按照提示完成安裝。
問:我如何完成 PyTorch 安裝?
答:從 PyTorch 網站複製安裝命令並在 PowerShell 中執行。
問:我如何驗證 PyTorch 是否已正確安裝?
答:在 PowerShell 中運行 'python',然後 'import torch',並使用 'torch.cuda.is_available()' 進行檢查。

分享至

DICloak防關聯指紋瀏覽器-防止賬號封禁,安全管理多帳號

讓多賬號運營更簡單高效,低成本高速度實現業務增長

相關文章