当前位置:   article > 正文

将CPU版本的pytorch换为GPU版本_怎么把cpu的torch变成gpu的

怎么把cpu的torch变成gpu的

     我发现用上文中的安装方式,安装的pytorch版本为CPU版的,在运行程序时很卡,很影响电脑性能。

        一、查看pytorch版本:首先进入python,

        import torch

        print(torch.__version__)

显示结果:2.2.1+cpu

        不用考虑卸载之前版本的pytorch,在安装gpu版本时,系统会自动删除cpu版本

        不建议使用镜像安装,我之前安装都不成功,直接使用pytorch官网生成的命令就可以啦

        二、直接安装GPU版本的pytorch

        conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

        这里安装pytorch要和cuda版本相匹配,若cuda版本是11.3,直接将cudatoolkit=10.1改为cudatoolkit=11.3即可。

        安装成功再次查看pytorch版本不显示CPU,显示结果为:2.2.1


 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/851427
推荐阅读
相关标签
  

闽ICP备14008679号