2024-03-29 10:29:23 发布
网友
PyTorch installation web page显示如何安装Pythorch的GPU和CPU版本:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
以及
两个版本可以安装在同一个Conda环境中吗?在
Conda
如果你可能会问为什么需要这个,那是因为我想要一个单一的Conda环境,我可以在有GPU和没有GPU的计算机上使用
GPU版本的PyTorch实际上是CPU PyTorch的超集。可以在CPU上使用GPU Pythorch,但不能在GPU上使用CPU Pythorch。所以在您的例子中,只安装GPU版本的PyTorch就足够了。在
GPU版本的PyTorch实际上是CPU PyTorch的超集。可以在CPU上使用GPU Pythorch,但不能在GPU上使用CPU Pythorch。所以在您的例子中,只安装GPU版本的PyTorch就足够了。在
相关问题 更多 >
编程相关推荐