如何解决.cuda()加载用时很长的问题
更新时间：2021年05月24日 14:50:06 作者：wang xiang
这篇文章主要介绍了如何解决.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
即可。
补充：关于win10环境下在使用torch调用cuda（）失败的原因
看了无数篇博客都是牛头不对马嘴，忙活了半天才搞清楚
如果大家在用torch的过程中看见这样的错误。。。。淡定，不是你的问题，也不是代码问题。。。是破win10的原因
只需要在代码前面加上
torch.cuda.current_device() torch.cuda._initialized = True
就可以正常训练了。。。心累啊
相关文章
这篇文章主要介绍了在Python 中将类对象序列化为JSON，序列化是将对象转换为可以在以后保存和检索介质中的过程，下文具体的内容分享,需要的朋友可以参考一下2022-04-04
这篇文章主要介绍了Python通用循环的构造方法,结合实例形式分析了Python常见的交互循环、哨兵循环、文件循环、死循环等实现与处理技巧,需要的朋友可以参考下2018-12-12
今天小编就为大家分享一篇Tensorflow分批量读取数据教程，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
比起matplotlib，pyeacharts的图表要丰富而且好看，而且pyechart文档全，便于开发和阅读文档，熟练掌握后是一种非常好用的数据可视化的工具之一。本文就来用Pyecharts绘制树形图，快跟随小编一起学习一下吧2023-02-02
本文主要介绍了python主动抛出异常raise的方法实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2023-12-12
这篇文章主要介绍了python插入数据到列表的方法,涉及Python中insert方法的使用技巧,非常具有实用价值,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了python 排序算法总结及实例详解的相关资料,需要的朋友可以参考下2016-09-09
Python3 Tensorlfow:增加或者减小矩阵维度的实现
这篇文章主要介绍了Python3 Tensorlfow:增加或者减小矩阵维度的实现，具有好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
这篇文章主要介绍了Python Mock模块原理及使用方法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-07-07
这篇文章主要介绍了使用python处理一万份word表格简历操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
最新评论