深度学习
文章平均质量分 51
污喵王z
这个作者很懒,什么都没留下…
展开
-
【pytorch】错误:No module named ‘typing_extensions‘ 问题解决
前几天跑模型发现报了一个很奇怪的错,点进去发现是源码里面的代码报错,这就有点离谱了。。。在第34行报错,如图:源代码文件位置在你conda虚拟环境下面类似于这种:…\envs\py3.7\Lib\site-packages\torch\ _jit_internal.py注意文件名叫:_jit_internal.py读代码了解到,python3.7及以下版本是需要用到 typing_extensions.py 这个文件中的代码的,而我所使用的python版本正好是3.7,所以出现这个错误的原因就是原创 2021-12-25 23:47:07 · 35506 阅读 · 18 评论 -
【Pytorch】内存够用但还是 RuntimeError: CUDA out of memory 问题解决
这里简述一下我遇到的问题:可以看到可用内存是大于需要被使用的内存的,但他依旧是报CUDA out of memory的错误我的解决方法是:修改num_workers的值,把它改小一点,就行了,如果还不行可以考虑使用以下方法:1.减小batch_size2.运行torch.cuda.empty_cache()函数3.释放GPU内存,kill掉没用的进程4.更改使用的显卡。这里大多指在服务器上运行,没有制定显卡,都默认使用第一张,或者多人使用,产生的不足。参考链接:https://blog.c原创 2021-12-25 23:06:55 · 4293 阅读 · 1 评论