使用pip命令pip uninstall torch
就可以卸载当前的torch版本。(不是cpu版本一般也没有必要重装吧?)
接着找到官网https://pytorch.org/get-started/locally/
在里面选择
根据你自己的需要选择最新的(我现在是11.8)或者没那么新的(11.7)的GPU版torch
接着复制下面的pip3的安装命令pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
这个命令是根据你上面选择的哪些红色选项而生成的。
等运行完成该命令即可成功安装gpu版本torch文章来源地址https://www.toymoban.com/news/detail-544174.html
文章来源:https://www.toymoban.com/news/detail-544174.html
到了这里,关于怎么把pytorch从CPU版本替换成GPU版本的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!