model.cuda()方法加载时间长怎么,如何解决好

Admin 2022-07-23 群英技术资讯 714 次浏览

这篇文章将为大家详细讲解有关“model.cuda()方法加载时间长怎么,如何解决好”的知识,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。


解决方法之一:

如果pytorch在进行model.cuda()操作需要花费的时间很长,长到你怀疑GPU的速度了,那就是不正常的。

如果你用的pytorch版本是0.3.0,升级到0.3.1就好了!

.cuda()加载时间很长的其他解决方法

方法一:

pip install --upgrade --force-reinstall http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

后面这个可以自己到官网上找自己对应的python和cuda版本的安装包,

官网为链接:

方法二:

conda install pytorch torchvision cuda80 -c soumith

这里可以根据自己cuda版本确定。查看cuda版本,在python命令行里面输入:

import torch
torch.version.cuda

即可。


上述内容具有一定的借鉴价值,感兴趣的朋友可以参考,希望能对大家有帮助,想要了解更多"model.cuda()方法加载时间长怎么,如何解决好"的内容,大家可以关注群英网络的其它相关文章。 群英智防CDN,智能加速解决方案

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:mmqy2019@163.com进行举报,并提供相关证据,查实之后,将立刻删除涉嫌侵权内容。

猜你喜欢

成为群英会员,开启智能安全云计算之旅

立即注册
专业资深工程师驻守
7X24小时快速响应
一站式无忧技术支持
免费备案服务
免费拨打  400-678-4567
免费拨打  400-678-4567 免费拨打 400-678-4567 或 0668-2555555
在线客服
微信公众号
返回顶部
返回顶部 返回顶部
在线客服
在线客服