windows下面通过conda安装torch和cuda的正确添加源

在windows下面需要安装alphapose的依赖环境,首先需要安装pytorch=1.1.0, torchvision=0.3.0, cudatoolkit=10.0, cudnn=7.6.5。

如果没有添加conda的镜像,则会提示找不到这些包。

0. conda换回默认的源

conda config --remove-key channels

1.  添加pytorch和torchvision的conda镜像,执行:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

2. 添加cudatoolkit的conda镜像,执行:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

3. 添加cudnn的conda镜像,执行:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

4. 最后执行命令使conda镜像生效:

conda config --set show_channel_urls yes

5. 执行安装命令:

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 cudnn=7.6.5

其中一般只执行0和1步骤,即可直接执行5步骤即可。

6. cudnn提示安装包找不到的情况

有时候比如想装cudnn = 8.0.5,但是执行 conda install cudnn=8.0.5, 却提示 “PackagesNotFoundError: The following packages are not available from current channels”,这个时候按照这篇进行处理

首先查看可用的cudnn版本,执行:anaconda search -t conda cudnn

出现如下图:

这个时候比如想装nvidia/cudnn这个版本,则执行:conda install -c https://conda.anaconda.org/nvidia cudnn,注意后面的/用空格代替

 

 即可装上满足cuda 11.0的cudnn。

 参考文献:

1. https://zhuanlan.zhihu.com/p/269972551

2. https://blog.csdn.net/just_h/article/details/90451935

3. https://www.cnblogs.com/yibeimingyue/p/13837929.html

 

 

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
你可以按照以下步骤来安装condatorchcuda和cudnn: 1. 首先,安装Anaconda或Miniconda,这是一个用于管理Python环境和软件包的工具。你可以从Anaconda官方网站(https://www.anaconda.com)下载适合你操作系统的版本,并按照官方文档的说明进行安装。 2. 安装CUDA(Compute Unified Device Architecture),这是一个用于利用NVIDIA GPU进行并行计算的平台。你可以从NVIDIA官方网站(https://developer.nvidia.com/cuda-downloads)下载适合你操作系统和GPU的CUDA版本,并按照官方文档的说明进行安装。 3. 安装cuDNN(CUDA Deep Neural Network library),这是NVIDIA提供的针对深度神经网络的加速库。你需要先注册一个NVIDIA开发者账号,然后从NVIDIA开发者网站(https://developer.nvidia.com/cudnn)下载适合你CUDA版本的cuDNN,并按照官方文档的说明进行安装。 4. 创建一个新的conda环境,并安装PyTorch。你可以使用以下命令创建一个名为"myenv"的新环境: ``` conda create --name myenv ``` 接下来,激活这个环境: ``` conda activate myenv ``` 然后,安装PyTorch和相应的CUDA版本,比如PyTorch 1.9.0: ``` conda install pytorch==1.9.0 torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch ``` 记得将`<your_cuda_version>`替换为你安装CUDA版本,比如`10.2`。 现在,你已经成功安装condatorchcuda和cudnn,并创建了一个新的conda环境,其中包含了PyTorch和相关依赖。你可以在这个环境中进行深度学习任务了。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

鸡啄米的时光机

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值