1. 程式人生 > >tensorflow GPU視訊記憶體佔滿而Util為0的問題

tensorflow GPU視訊記憶體佔滿而Util為0的問題

問題1:Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA

 解決:程式碼中新增

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

問題2:Tensorflow GPU視訊記憶體佔滿,而Util為0

解決:程式碼中新增

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "8" //這裡指定GPU8

程式碼中同時指定了GPU,避免了其他GPU的視訊記憶體佔用。