pytorch-windows安装-pytorch在windows

pytorch-windows安装-pytorch在windows

pytorch windows安装

以下是在 Windows 系统上安装 PyTorch 的一般步骤:

pytorch-windows安装-pytorch在windows

安装 Anaconda


  1. 访问Anaconda 官网。
  2. 选择适合 Windows 系统的 Anaconda 安装包进行下载,一般建议选择 Python 3.x 版本对应的 Anaconda3。也可以通过清华镜像地址下载,速度可能更快:清华 Anaconda 镜像源。
  3. 下载完成后,点击运行安装包,点击 “Next”。
  4. 选择安装路径,建议选择默认安装路径,也可自行指定,但路径不要包含中文。
  5. 勾选 “Add Anaconda to my PATH environment variable” 和 “Register Anaconda as the system Python 3.x”。
  6. 等待安装完成。

创建虚拟环境


  1. 打开 “Anaconda Prompt”。
  2. 创建新的虚拟环境,执行命令:conda create -n pytorch_env python=3.11,可根据需求更改 Python 版本。
  3. 激活虚拟环境,执行命令: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


  1. 访问PyTorch 官网。
  2. 根据电脑的实际情况选择配置,如操作系统为 Windows,选择对应的 Python 版本、CUDA 版本等。
  3. 官网会生成相应的安装命令,有condapip两种方式。
    • 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


验证安装


在 “Anaconda Prompt” 中进入虚拟环境后,执行以下命令:
收起

python

python torch torchcudais_available
如果返回True,表示安装成功且 GPU 可用;若返回False,可能安装的是 CPU 版本或 GPU 相关配置有问题。也可以通过print(torch.__version__)输出 PyTorch 的版本号来验证安装。

看过该文章的人还看了