以下是在 Windows 系統(tǒng)中安裝 PyTorch 的詳細(xì)步驟：

1. 安裝 Anaconda（可選但推薦）

• Anaconda 是一個(gè)用于科學(xué)計(jì)算的 Python 發(fā)行版，它可以幫助你更方便地管理 Python 環(huán)境和安裝各種庫。
• 你可以從 Anaconda 官方網(wǎng)站（https://www.anaconda.com/products/individual）下載適合你 Windows 系統(tǒng)版本（32 位或 64 位）的 Anaconda 安裝包。
• 下載完成后，雙擊安裝包，按照安裝向?qū)У奶崾具M(jìn)行安裝。在安裝過程中，你可以選擇安裝路徑等選項(xiàng)。

2. 創(chuàng)建并激活虛擬環(huán)境（可選但推薦）

• 打開 Anaconda Prompt（如果沒有安裝 Anaconda，可以使用 Windows 的命令提示符）。
• 創(chuàng)建一個(gè)新的虛擬環(huán)境，例如，命名為 “pytorch_env”，在 Anaconda Prompt 中輸入以下命令：
  • conda create -n pytorch_env python=3.8（這里以 Python 3.8 為例，你可以根據(jù)需要選擇其他版本）
  
  
• 激活虛擬環(huán)境，輸入：
  • 在 Anaconda Prompt 中：conda activate pytorch_env
  • 如果使用系統(tǒng)自帶的命令提示符，需要先找到虛擬環(huán)境所在的路徑，然后使用該路徑下的activate.bat文件來激活，例如：C:\Users\YourUsername\Anaconda3\envs\pytorch_env\Scripts\activate.bat
  
  

3. 確定你的系統(tǒng)和計(jì)算平臺(tái)

• PyTorch 的安裝需要根據(jù)你的系統(tǒng)（Windows）和計(jì)算平臺(tái)（CPU 或 GPU）來選擇合適的安裝命令。
• 如果你的電腦沒有 NVIDIA GPU，或者你只想使用 CPU 來運(yùn)行 PyTorch，那么安裝會(huì)相對(duì)簡單。
• 如果有 NVIDIA GPU 并且想利用 GPU 加速計(jì)算，你需要先安裝合適的 NVIDIA 驅(qū)動(dòng)程序和 CUDA 工具包。

4. 安裝 PyTorch（CPU 版本）

• 如果你只想使用 CPU，在激活的虛擬環(huán)境中，使用以下命令安裝 PyTorch：
  • pip install torch torchvision torchaudio
  • 這個(gè)命令會(huì)從 PyTorch 官方的 Python 包索引中下載并安裝最新的 CPU 版本的 PyTorch 及其相關(guān)庫（torchvision 用于計(jì)算機(jī)視覺任務(wù)，torchaudio 用于音頻處理）。
  
  

5. 安裝 PyTorch（GPU 版本）

• 首先，確保你的 NVIDIA GPU 支持 CUDA，并且已經(jīng)安裝了合適的 CUDA 工具包。一般來說，你需要安裝與 PyTorch 兼容的 CUDA 版本。
• 然后，根據(jù) CUDA 版本選擇合適的安裝命令。例如，對(duì)于 CUDA 11.8，在激活的虛擬環(huán)境中，使用以下命令：
  • pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
  
  
• 安裝完成后，你可以通過以下簡單的代碼來驗(yàn)證 PyTorch 是否安裝成功并且可以使用 GPU（如果是安裝的 GPU 版本）：
  • 打開 Python 交互式環(huán)境（在激活虛擬環(huán)境后，輸入python），然后輸入以下代碼：
  
  

收起

python

復(fù)制

torch torch__version__ torchcudais_available device torchdevice device
這段代碼會(huì)打印出 PyTorch 的版本號(hào)，并且如果 GPU 可用，會(huì)顯示相應(yīng)的信息。