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

因为一些众所周知的原因,你安装cuda11.6+pytorch的时候是会很蛋疼的。于是我们需要修改一下cuda的软件源。

 

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

以上命令复制粘贴即可。

需要注意的一点是,官方安装pytorch给的命令是

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

我们需要修改一下,去掉 -c pytorch -c nvidia

conda install pytorch torchvision torchaudio pytorch-cuda=11.6

这就搞定了。

给TA打赏
共{{data.count}}人
人已打赏
站长笔记

revCHATGPT开发示例

2023-2-16 15:33:23

站长笔记

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

2023-2-20 20:53:17

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索