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

Question

PyTorch 安装 web 页面显示如何安装 GPU 和 CPU 版本的 Z95B88F180E9EB5678E0F9EBACZ2：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

和

conda install pytorch torchvision cpuonly -c pytorch

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

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

Answer 1

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