欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

pytorch model.cuda()花费时间很长的解决

时间:2021-07-23 07:38:06|栏目:Python代码|点击:

解决方法之一:

如果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

即可。

上一篇:基于Python pyecharts实现多种图例代码解析

栏    目:Python代码

下一篇:python 创建一维的0向量实例

本文标题:pytorch model.cuda()花费时间很长的解决

本文地址:http://www.codeinn.net/misctech/160538.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有