使用 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

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐