主页 > 知识库 > 如何解决.cuda()加载用时很长的问题

如何解决.cuda()加载用时很长的问题

热门标签:手机网页嵌入地图标注位置 天津电话机器人公司 400电话办理哪种 电销机器人的风险 地图标注线上如何操作 开封语音外呼系统代理商 应电话机器人打电话违法吗 开封自动外呼系统怎么收费 河北防封卡电销卡

方法一:

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

即可。

补充:关于win10环境下在使用torch调用cuda()失败的原因

看了无数篇博客都是牛头不对马嘴,忙活了半天才搞清楚

如果大家在用torch的过程中看见这样的错误。。。。淡定,不是你的问题,也不是代码问题。。。是破win10的原因

只需要在代码前面加上

torch.cuda.current_device()
torch.cuda._initialized = True

就可以正常训练了。。。心累啊

您可能感兴趣的文章:
  • pytorch查看torch.Tensor和model是否在CUDA上的实例
  • Pytorch 解决自定义子Module .cuda() tensor失败的问题
  • pytorch model.cuda()花费时间很长的解决

标签:驻马店 江苏 山东 兰州 宿迁 常州 六盘水 成都

巨人网络通讯声明:本文标题《如何解决.cuda()加载用时很长的问题》,本文关键词  如何,解决,.cuda,加载,用时,;如发现本文内容存在版权问题,烦请提供相关信息告之我们,我们将及时沟通与处理。本站内容系统采集于网络,涉及言论、版权与本站无关。
  • 相关文章
  • 下面列出与本文章《如何解决.cuda()加载用时很长的问题》相关的同类信息!
  • 本页收集关于如何解决.cuda()加载用时很长的问题的相关信息资讯供网民参考!
  • 企业400电话

    智能AI客服机器人
    15000

    在线订购

    合计11份范本:公司章程+合伙协议+出资协议+合作协议+股权转让协议+增资扩股协议+股权激励+股东会决议+董事会决议

    推荐文章