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

作者：天景科技苑 | 2024-07-19 12:28:10

踩

怎么换成gpu

使用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.wpsshop.cn/w/天景科技苑/article/detail/851490