nvida显卡使用

nvidia显卡使用

1. NVIDIA GPU 显存释放

显卡释放

# 指定第 0 号 GPU 进程占用
fuser -v /dev/nvidia0

# 指定所有 GPU 进程占用
fuser -v /dev/nvidia*

# 根据进程 PID 查看程序内容
sudo ls -l /proc/PID

# 如果是 PYTHON 程序，会有 exe，cwd 等信息
sudo ls -l /proc/PID | grep exe
sudo ls -l /proc/PID | grep cwd

# 如果查询发现该进程就是占用显存的进程，那么直接杀死
sudo kill -9 PID

# 清除进程后，再次查看显存占用
# 一般会有多个进程占用 GPU 显存，可以都查看下，是否需要杀死，然后在查看显存情况
watch -n 2 nvidia-smi

2. 指定使用显卡

CUDA_VISIBLE_DEVICES=1,2 python your_script.py # 指定1,2块使用