pytorch-windows安裝-pytorch在windows
pytorch windows安裝
以下是在 Windows 系統上安裝 PyTorch 的一般步驟:

安裝 Anaconda
- 訪問Anaconda 官網。
- 選擇適合 Windows 系統的 Anaconda 安裝包進行下載,一般建議選擇 Python 3.x 版本對應的 Anaconda3。也可以通過清華鏡像地址下載,速度可能更快:清華 Anaconda 鏡像源。
- 下載完成后,點擊運行安裝包,點擊 “Next”。
- 選擇安裝路徑,建議選擇默認安裝路徑,也可自行指定,但路徑不要包含中文。
- 勾選 “Add Anaconda to my PATH environment variable” 和 “Register Anaconda as the system Python 3.x”。
- 等待安裝完成。
創建虛擬環境
- 打開 “Anaconda Prompt”。
- 創建新的虛擬環境,執行命令:
conda create -n pytorch_env python=3.11,可根據需求更改 Python 版本。 - 激活虛擬環境,執行命令:
conda activate pytorch_env。
更換鏡像源(可選)
如果安裝過程中下載速度慢,可以更換鏡像源。
- pip 換源:在文件管理器地址欄輸入
%APPDATA%回車,進入C:\Users\電腦用戶\AppData\Roaming文件夾,新建pip文件夾并在其中新建pip.ini配置文件。用記事本打開pip.ini,輸入(global)index-url = http://pypi.douban.com/simple(install)use-mirrors =truemirrors =http://pypi.douban.com/simple/trusted-host =pypi.douban.com,保存文件。 - conda 換源:執行
conda config --set show_channel_urls yes生成.condarc文件。在C:\Users\電腦用戶下找到.condarc文件,打開后將原始內容刪除,將以下內容復制進去并保存:
收起
plaintext
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
安裝 PyTorch
- 訪問PyTorch 官網。
- 根據電腦的實際情況選擇配置,如操作系統為 Windows,選擇對應的 Python 版本、CUDA 版本等。
- 官網會生成相應的安裝命令,有
conda和pip兩種方式。- conda 方式:如
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia。 - pip 方式:如
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124。
- conda 方式:如
驗證安裝
在 “Anaconda Prompt” 中進入虛擬環境后,執行以下命令:
收起
python
python
torch
torchcudais_available
如果返回True,表示安裝成功且 GPU 可用;若返回False,可能安裝的是 CPU 版本或 GPU 相關配置有問題。也可以通過print(torch.__version__)輸出 PyTorch 的版本號來驗證安裝。
