在同一個Conda環境中可以安裝PyTorch的GPU版本和CPU版本嗎?

2024-03-29 10:29:23 发布

您现在位置:Python中文网/ 问答频道 /正文

PyTorch installation web page显示如何安装Pythorch的GPU和CPU版本:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

以及

^{2}$

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

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


Tags: install版本webgpu环境pageinstallationpytorch
1条回答
网友
1楼 · 发布于 2024-03-29 10:29:23

GPU版本的PyTorch实际上是CPU PyTorch的超集。可以在CPU上使用GPU Pythorch,但不能在GPU上使用CPU Pythorch。所以在您的例子中,只安装GPU版本的PyTorch就足够了。在

相关问题 更多 >