使用 conda 安装 cuda

使用 conda 安装 cuda 时,一共需要安装以下几个组件:

  1. cudatoolkit
  2. cuda-nvcc

遇到这个问题,一般是在安装指定版本(尤其是老版本的)pytorch 时,遇到的。

我们可以在 pytorch 的历史版本之中,找到 pytorch 对应的 cuda 和他组件的版本:

在下列连接之中,我们可以找到对应的版本。同时,我们可以看到他所支持的 cuda 版本有 11.8, 12.1, 12.4。


我们以下假设我们需要安装的是 11.8 版本的 cuda .

我们的安装的思路是,手动照 cuda-nvcc 的版本,并让 conda 自动搜寻合适的 cudatoolkit 的版本。

我们可以在 anaconda 的主页之中,来找到我们想要的 cuda-nvcc 版本:

anaconda 连接:https://anaconda.org/
 

一般选择官方源(对于 cuda,是 nvidia 或者 conda-forge):

这里可以得到具体的版本号,cuda-nvcc 的 11.8 的版本有 11.8.89

可以通过以下链接进行下载(这里 -c 表示选择的是什么频道,因为是 nvidia 频道的,所以如此填写):

conda install -c nvidia cuda-nvcc=11.8.89 cudatoolkit

或者可以缩略写

conda install -c cuda-nvcc=11.8 cudatoolkit

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值