logo

PyTorch:轻松安装与切换清华源镜像指南

作者:搬砖的石头2023.09.27 14:00浏览量:159

简介:在 Windows 系统上,使用 Conda 安装 PyTorch 并切换到清华源镜像是一种高效的方式。本教程将指导您完成以下步骤:

在 Windows 系统上,使用 Conda 安装 PyTorch 并切换到清华源镜像是一种高效的方式。本教程将指导您完成以下步骤:

  1. 安装 Anaconda
    首先,您需要安装 Anaconda。可以在官网下载最新版本的 Anaconda,并按照安装向导进行安装。安装完成后,启动 Anaconda Navigator 并更新 conda 工具。
  2. 创建 Conda 环境
    使用以下命令创建一个新的 Conda 环境,名称可以自由指定,例如:pytorch_env。
    1. conda create -n pytorch_env python=3.7
  3. 激活 Conda 环境
    使用以下命令激活刚刚创建的环境:
    1. conda activate pytorch_env
  4. 安装 PyTorch
    使用以下命令在 Conda 环境中安装 PyTorch:
    1. conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch
    其中,cudatoolkit=xx.x 中的 xx.x 表示您的 GPU 对应的 CUDA 版本。例如,如果您的 GPU 支持 CUDA 10.1,则可以使用 cudatoolkit=10.1。如果不确定您的 GPU 支持的 CUDA 版本,请查看 PyTorch 官方文档或 CUDA 文档。
  5. 切换清华源镜像
    使用以下命令将 Conda 渠道切换到清华源:
    1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/conda-forge
    2. conda config --set channel_priority strict
    执行完上述命令后,您就可以使用清华源的 PyTorch 镜像了。可以通过执行以下命令查看当前 Conda 配置的渠道列表:
    1. conda config --show channels
  6. 更新 PyTorch
    如果您需要更新 PyTorch 版本,可以使用以下命令:
    1. conda update pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch_channel_name
    其中,pytorch_channel_name 是您指定的 PyTorch 渠道名称。在本例中,我们使用清华源的 PyTorch 镜像,因此可以将其替换为 pytorch_tuna。如果您已经将渠道切换到清华源,则不需要再指定渠道名称。
    总结:本教程介绍了在 Windows 系统上使用 Conda 安装 PyTorch 并切换到清华源镜像的方法。通过这种方法,您可以轻松地安装 PyTorch 并更新版本,同时避免了繁琐的配置过程。希望对您有所帮助!

相关文章推荐

发表评论