.cuda()加载用时很长解决方法

本篇文章重点为大家讲解一下决.cuda()加载用时很长的问题，有需要的小伙伴可以参考一下。

姑苏网站制作公司哪家好，找成都创新互联公司！从网页设计、网站建设、微信开发、APP开发、自适应网站建设等网站项目制作，到程序开发，运营维护。成都创新互联公司从2013年成立到现在10年的时间，我们拥有了丰富的建站经验和运维经验，来保证我们的工作的顺利进行。专注于网站建设就选成都创新互联公司。

方法一：

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

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

当前文章：.cuda()加载用时很长解决方法
转载源于：http://www.stwzsj.com/qtweb/news17/1467.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联