安装后PyTorch版本不正确

kjthegm6 于 2023-11-19 发布在其他

关注(0)|答案(1)|浏览(139)

screen我在Windows 11上安装PyTorch，使用此命令pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121但安装后发现版本安装为CPU，而不是CUDA，正如日志中所说的那样。令人惊讶的是，torchaudio版本是2.1.0+cu121，但torch和torchvision与截图相同，所以我不能使用CUDA，也许有人知道如何解决这个问题
我已经：
1.更改Cuda版本
1.手动安装.whl
1.重新安装的pytorch
UPD：我运行了pip3 list，发现它看到它是正确的，但我不知道为什么它在notebook中使用不同的版本

pytorch

来源：https://stackoverflow.com/questions/77445653/incorrect-version-of-pytorch-after-installation