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

Question

[C0显示如何安装PyTorch的GPU和CPU版本：

PyTorch installation web page

和

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

两个版本都可以安装在相同的conda install pytorch torchvision cpuonly -c pytorch 环境中吗？

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

Answer 1

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