【DeepLearning笔记】Tensorflow、Torch指定GPU运行与linux显存、内存分析

Tensorflow指定GPU

硬性指定GPU服务器

只使用第三块GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

禁止使用GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

动态分配显存大小

让TensorFlow按需分配显存：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True

直接指定显存比例：

config.gpu_options.per_process_gpu_memory_fraction = 0.4
#占用GPU40%

linux显存、内存分析

在运行程序时常常遇见显存移除、内存升满等情况

查看显存使用情况

naidia-smi

查看内存使用情况

ps aux --sort -rss

查询进程的运行用户及地址

ll /proc/**

以上命令基本可以满足我们日常的需要：）