
深度学习
焦志坤
一名程序猿的自我救赎之路
展开
-
RuntimeError: CUDA out of memory解决方法
RuntimeError: CUDA out of memory解决方法 前言 今天在运行代码的时候出现了cuda的一个报错,报错如下,意思就是超出内存了 解决 首先查看一下GPU使用情况,命令如下:nvidia-smi 看输出的第二列(Memory_Usage)查看各个GPU使用情况 找到剩余内存较大的GPU,然后代码中输入如下代码 import os import torch os.environ['CUDA_VISIBLE_DEVICES'] = '2,6' # 代表可以使用第二个和第六个原创 2021-12-02 09:55:44 · 24011 阅读 · 1 评论 -
python调用TensorFlow时出现FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecate
python调用TensorFlow时出现FutureWarning: Passing (type, 1) or ‘1type’ as a synonym of type is deprecate 点进警告的文件: 修改成原创 2019-09-23 10:48:21 · 469 阅读 · 1 评论 -
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2如何隐藏
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2如何隐藏 import os.environ[‘TF_CPP_MIN_LOG_LEVEL’]=‘2’ log 信息分为四个等级,按照重要性程度递增为: INFO(通知) < WARNING(警告) < ERRO...原创 2019-09-23 11:04:25 · 162 阅读 · 0 评论