tyty-Somnuspoppy

一、命令行运行python程序时

1、首先查看哪些GPU空闲,nvidia-smi显示当前GPU使用情况。

nvidia-smi

2、然后指定空闲的GPU运行python程序。

CUDA_VISIBLE_DEVICES=0,2,3 python test.py

二、在python程序中指定GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3"

PS:周期性输出GPU使用情况

每 10s 显示一次GPU使用情况:

watch -n 10 nvidia-smi

   

分类:

技术点:

相关文章:

  • 2022-01-17
  • 2021-09-09
  • 2022-12-23
  • 2022-12-23
  • 2022-01-01
  • 2021-07-27
  • 2021-04-01
猜你喜欢
  • 2021-07-18
  • 2021-08-12
  • 2022-12-23
  • 2022-12-23
  • 2021-05-20
  • 2021-06-04
相关资源
相似解决方案