怎么把pytorch从CPU版本替换成GPU版本

使用pip命令pip uninstall torch就可以卸载当前的torch版本。(不是cpu版本一般也没有必要重装吧?)
接着找到官网https://pytorch.org/get-started/locally/
在里面选择怎么把pytorch从CPU版本替换成GPU版本_第1张图片
根据你自己的需要选择最新的(我现在是11.8)或者没那么新的(11.7)的GPU版torch
接着复制下面的pip3的安装命令pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118这个命令是根据你上面选择的哪些红色选项而生成的。
等运行完成该命令即可成功安装gpu版本torch

你可能感兴趣的:(pytorch,人工智能,python,torch)