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

【问题标题】：Can I install pytorch cpu + any specified version of cudatoolkit?我可以安装 pytorch cpu + 任何指定版本的 cudatoolkit 吗？
【发布时间】：2021-12-13 06:16:18
【问题描述】：

我的遥控器有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

我发现pytorch 是一个cpu。

或者，我将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.0 和cuda==11.0？

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

【问题讨论】：

这只是一个疯狂的猜测，但我认为您看到的行为是因为 pytorch 频道没有针对 cudatoolkit 11.0 构建的 pytorch 版本。你可以检查这个，例如通过限制构建字符串：conda install "pytorch=*=*cuda11.0*" -c pytorch -c conda-forge
这就是我的观点。我想安装一个基于cudatoolkit==11.0 的pytorch==1.10.0。我想知道将pytorch==1.10.0 (cpu) 和cudatoolkit==11.0 一起安装是否安全。
我怀疑如果这样做，您将无法将模型移动到 GPU 设备，但会出现某种错误。但我完全不确定，所以请谨慎对待。
我试过了。而且似乎一切正常...我不知道是否存在一些潜在的错误。
对不起，我发现安装cpu torch + cudatoolkit 不行。我之前错了。

标签： pytorch conda

【解决方案1】：

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

如 cmets 中所述，可以尝试强制使用 CUDA 构建 PyTorch

conda create -n foo -c pytorch -c conda-forge cudatoolkit=11.0 'pytorch=*=*cuda*'

在这种组合中会失败。

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

【讨论】：

谢谢你。