• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    pytorch 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
    

    即可。

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

    您可能感兴趣的文章:
    • pytorch 如何用cuda处理数据
    • pytorch中.to(device) 和.cuda()的区别说明
    • PyTorch CUDA环境配置及安装的步骤(图文教程)
    • Linux安装Pytorch1.8GPU(CUDA11.1)的实现
    • 将pytorch的网络等转移到cuda
    上一篇:如何理解及使用Python闭包
    下一篇:Python函数装饰器的使用教程
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

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