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