常见问题

 

如何查看深度学习平台的 GPU 使用率？

更新时间 2021-02-16

  • 在容器或主机中运行 nvidia-smi 查看 GPU 设备是否正常，也可以使用 watch -n 0.5 nvidia-smi 动态的查看 GPU 的使用情况。

  • 查看 TensorFlow 是否可以正常使用 GPU

import  tensorflow as tf
sess = tf.InteractiveSession()

Alt text

  • 查看 Keras 是否可以正常使用 GPU
from keras import backend as K
K.tensorflow_backend._get_available_gpus()

Alt text

  • 查看 Caffe 是否可以正常使用 GPU
cd /root/caffe
sh data/mnist/get_mnist.sh
sh examples/mnist/create_mnist.sh
sed -i "s/solver_mode: CPU/solver_mode: GPU/g" examples/mnist/lenet_solver.prototxt
sh examples/mnist/train_lenet.sh

Alt text

  • 查看 PyTorch 是否可以正常使用 GPU
import torch
torch.cuda.is_available()
torch.cuda.current_device()
torch.cuda.device(0)
torch.cuda.device_count()
torch.cuda.get_device_name(0)

Alt text

这篇文档解决了您的问题吗？
0
0