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版本就足够了。