在安裝 PyTorch 之前,請確保您的計算機上已安裝 Python 和 pip。如果您尚未安裝它們,請參考可靠的在線資源以獲取設置指南。當您準備好 Python 和 pip 後,便可以繼續安裝 PyTorch。
要開始安裝過程,請訪問官方 PyTorch 網站。向下滾動直到找到安裝選項。選擇穩定版本的 PyTorch,目前是版本 2.1.1。選擇您的操作系統;例如,如果您使用的是 Windows,請選擇該選項。由於您沒有安裝 Conda,因此將包留作 pip,並確保選擇 Python 作為語言。
計算平台選項至關重要,因為它與您的顯示卡有關。如果您有 Nvidia 顯示卡,請選擇相應的 CUDA 版本。對於 AMD 顯示卡,選擇 ROCm。如果您的系統沒有 GPU,則選擇 CPU 選項。請注意,AMD 選項不適用於 Windows 用戶;它需要 Linux 操作系統。
在執行 pip install 命令之前,檢查系統上安裝的 CUDA 版本非常重要。打開 PowerShell 並輸入命令 'nvcc --version'。如果未安裝 CUDA,您需要從 Nvidia 網站下載它。
要安裝 CUDA,請在瀏覽器中搜索 'install CUDA 12.1' 並點擊 Nvidia 官方網站的第一個結果。選擇您的操作系統、架構(x86 64)和 Windows 版本。選擇 'exe local' 安裝程序類型,並確保在開始下載之前有至少 3.1 GB 的可用空間。
下載完成後,導航到您的下載文件夾並打開可執行文件。您可能需要刷新頁面以查看下載。雙擊可執行文件並允許其通過用戶帳戶控制。選擇提取路徑並繼續提取。按照提示操作,同意許可協議並選擇快速安裝選項。
在 CUDA 安裝過程中,您可能會遇到一條消息,指示需要支持的 Visual Studio 版本。為了解決此問題,請在瀏覽器中打開新標籤並搜索 Visual Studio。下載免費的社區版。下載完成後,打開安裝程序並選擇您希望安裝的包,確保包括 Python 開發包。
安裝 Visual Studio 後,返回 CUDA 安裝過程。再次打開 CUDA 12.1 可執行文件,選擇提取路徑,並繼續安裝。按照提示完成安裝過程。完成後,通過在 PowerShell 中運行 'nvcc --version' 來驗證安裝,以確認 CUDA 12.1 現在已安裝。
CUDA 安裝成功後,您現在可以返回 PyTorch 安裝過程。複製 PyTorch 網站上提供的安裝命令並將其粘貼到 PowerShell 中。執行該命令以在您的 Windows 10 或 11 操作系統上安裝 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()' 進行檢查。