关于国内conda安装cuda11.6+pytorch的那些事。

在众所周知的情况下,安装CUDA 11.6以及PyTorch可能会让人感到非常繁琐。幸运的是,我们可以通过修改软件源来解决这个问题。本教程将向您展示如何轻松地修改CUDA和PyTorch的软件源,以便顺利完成安装。

起始故事

在开启这个教程之前,让我与您分享一个故事。假设您是一名热衷于深度学习和人工智能的开发者,准备在您的电脑上安装CUDA 11.6和PyTorch,以便进行深度学习的研究和开发。然而,您可能已经意识到,由于一些众所周知的原因,官方安装命令可能会导致问题。不用担心,本教程将向您展示如何通过修改软件源来解决这个问题,让您可以顺利安装所需的工具,继续您的深度学习之旅。

修改软件源的步骤

下面是修改CUDA和PyTorch软件源的详细步骤:

步骤1:配置Conda的软件源

首先,打开终端并运行以下命令,将清华大学的镜像源添加到Conda的配置中:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --set show_channel_urls true

这些命令会将清华大学的镜像源添加到您的Conda配置中,以便后续安装时可以从这些源中获取所需的软件包。

步骤2:安装PyTorch

接下来,我们将安装PyTorch。请注意,官方安装PyTorch的命令可能包含-c pytorch -c nvidia参数,但在这里我们将其去掉,因为我们已经配置了清华大学的镜像源。

运行以下命令来安装PyTorch:

conda install pytorch torchvision torchaudio pytorch-cuda=11.6

这个命令将下载并安装PyTorch及其相关的库,而无需担心官方源可能会出现的问题。

步骤3:完成安装

恭喜您,您已经成功修改了CUDA和PyTorch的软件源,并且顺利安装了PyTorch。现在,您可以继续进行深度学习的研究和开发,无需担心安装问题。

总结

通过本教程,您学会了如何修改CUDA和PyTorch的软件源,以解决安装问题。这将帮助您顺利进行深度学习项目,无论您是一名开发者还是研究人员,都能受益于这些技巧。

声明:本站所有文章,如无特殊说明或标注,均为本站(王大神)原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。

给TA打赏
共{{data.count}}人
人已打赏
指数词

revCHATGPT开发示例

2023-2-16 15:33:23

指数词

如何在 Windows 上通过 Anaconda 安装 PyTorch 及 CUDA 环境

2023-2-20 20:53:17

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索