使用 conda 安装 cuda
在下列连接之中,我们可以找到对应的版本。同时,我们可以看到他所支持的 cuda 版本有 11.8, 12.1, 12.4。我们的安装的思路是,手动照 cuda-nvcc 的版本,并让 conda 自动搜寻合适的 cudatoolkit 的版本。这里可以得到具体的版本号,cuda-nvcc 的 11.8 的版本有 11.8.89。遇到这个问题,一般是在安装指定版本(尤其是老版本的)pytorch 时
使用 conda 安装 cuda 时,一共需要安装以下几个组件:
- cudatoolkit
- 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

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