PyTorch 的 GPU 和 CPU 版本可以安装在同一个 Conda 环境中吗？

Question

PyTorch installation web page 展示了如何安装 GPU 和 CPU 版本的 PyTorch：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

和

conda install pytorch torchvision cpuonly -c pytorch

这两个版本可以安装在同一个Conda环境中吗？

如果您可能会问为什么需要这样做，那是因为我想要一个单一的 Conda 环境，我可以在有 GPU 和没有 GPU 的计算机上使用它。

Answer 1

GPU 版本的 PyTorch 实际上是 CPU PyTorch 的超集。您可以在 CPU 上使用 GPU PyTorch，但不能在 GPU 上使用 CPU PyTorch。所以在你的情况下，只安装 GPU 版本的 PyTorch 就足够了。