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 的版本号来验证安装。
