pytorch安装GPU版本的torch，torchvision_0.16.2+cpu torchvision 是gpu吗

作者：我家自动化 | 2024-07-13 04:46:26

踩

0.16.2+cpu torchvision 是gpu吗

安装GPU版本的torch，利用pytorch官网的pip和conda命令下载的不是GPU版本的torch，torch后面没有+cu118.

自己使用国内镜像源安装指定版本的torch，torchvision，torchaudio

下面展示一些 内联代码片。

pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package --trusted-host mirrors.aliyun.com
1

注意torch版本和torchvision版本一定要匹配，torchaudio和torch一样就可以。cu版本也要统一。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家自动化/article/detail/817701