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

Question

PyTorch installation web page显示了如何安装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

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