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

Question

该PyTorch安装网页显示了如何安装PyTorch的GPU和CPU版本：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

和

conda install pytorch torchvision cpuonly -c pytorch

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

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

Answer 1

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