cuda()加载时间长有几种解决方法,怎样操作
Admin 2022-07-09 群英技术资讯 732 次浏览
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
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:mmqy2019@163.com进行举报,并提供相关证据,查实之后,将立刻删除涉嫌侵权内容。
猜你喜欢
这篇文章主要介绍了python中的metaclass详情,在python中的metaclass就是帮助developer实现元编程,更多详细内容需要的小伙伴可以参考一下
Altair被称为是统计可视化库,因为它可以通过分类汇总、数据变换、数据交互、图形复合等。本文和大家聊一下Python当中的altair可视化模块,并且通过调用该模块来绘制一些常见的图表,感兴趣的可以了解一下
众所周知Django较为适合原生开发,即通过该框架搭建一个全新的项目,通过在修改models.py来创建新的数据库表,下面这篇文章主要给大家介绍了关于django连接Mysql中已有数据库的相关资料,需要的朋友可以参考下
这篇文章主要介绍了nditer―numpy.ndarray 多维数组的迭代操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
程序设计的好与坏,早在我们青葱岁月时就接触过了,只是那是并不知道这竟如此重要。能够立即改善程序设计、写出“好”代码的知识有以下几点
成为群英会员,开启智能安全云计算之旅
立即注册Copyright © QY Network Company Ltd. All Rights Reserved. 2003-2020 群英 版权所有
增值电信经营许可证 : B1.B2-20140078 粤ICP备09006778号 域名注册商资质 粤 D3.1-20240008