我可以安装 pytorch cpu + 任何指定版本的 cudatoolkit 吗?

Can I install pytorch cpu + any specified version of cudatoolkit?

我的遥控器有 cuda==11.0,我想在上面安装 pytorch

我使用命令conda install pytorch cudatoolkit=11.0 -c pytorch -c conda-forge但是在安装列表中:

cudatoolkit        conda-forge/linux-64::cudatoolkit-11.0.3-h15472ef_8
pytorch            pytorch/linux-64::pytorch-1.10.0-py3.8_cpu_0

我发现 pytorchcpu 一个。

或者,我将 11.0 替换为 11.1,安装列表显示为:

cudatoolkit        conda-forge/linux-64::cudatoolkit-11.1.1-h6406543_8
pytorch            pytorch/linux-64::pytorch-1.10.0-py3.8_cuda11.1_cudnn8.0.5_0

其中 pytorch 是一个 gpu

我的问题是:以上两种安装方式本质上是一样的吗?如果没有,我如何安装 pytorch=1.10.0cuda==11.0

我还想知道 cuda 兼容性如何工作? cudatoolkit==11.1 是否与使用 cudatoolkit==11.0 编译的程序兼容?

这完全取决于 pytorch 频道是否针对特定 cudatoolkit 版本构建了一个版本。我不知道具体的搜索方式,但可以在 pytorch 频道上 browse what builds are available。对于 linux-64 平台上的 PyTorch 1.10,似乎只有 CUDA 版本 10.2、11.1 和 11.3 可用。

如评论中所述,可以尝试使用

强制 PyTorch 的 CUDA 构建
conda create -n foo -c pytorch -c conda-forge cudatoolkit=11.0 'pytorch=*=*cuda*'

在这个组合中会失败。

至于兼容性,不,pytorch 包在 cudatoolkit 的次要版本中构建锁。例如,