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

发表评论
登录后可评论,请前往 登录 或 注册