深度学习
焦志坤
一名程序猿的自我救赎之路
展开
-
RuntimeError: CUDA out of memory解决方法
RuntimeError: CUDA out of memory解决方法前言今天在运行代码的时候出现了cuda的一个报错,报错如下,意思就是超出内存了解决首先查看一下GPU使用情况,命令如下:nvidia-smi看输出的第二列(Memory_Usage)查看各个GPU使用情况找到剩余内存较大的GPU,然后代码中输入如下代码import osimport torchos.environ['CUDA_VISIBLE_DEVICES'] = '2,6' # 代表可以使用第二个和第六个原创 2021-12-02 09:55:44 · 23861 阅读 · 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 · 448 阅读 · 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如何隐藏importos.environ[‘TF_CPP_MIN_LOG_LEVEL’]=‘2’log 信息分为四个等级,按照重要性程度递增为:INFO(通知) < WARNING(警告) < ERRO...原创 2019-09-23 11:04:25 · 136 阅读 · 0 评论