勇敢地禁用 GPU Linux - Shell-Bash

在日常的开发工作中，我们经常需要使用 GPU 来加速计算，但有些时候我们需要禁用 GPU。本文将介绍如何以 Shell-Bash 的方式勇敢地禁用 GPU。

为什么要禁用 GPU

1. 测试 CPU 性能
2. GPU 被其他进程占用，导致当前进程受影响
3. 某些程序可能出现兼容性问题，需要禁用 GPU

禁用 GPU 的方式

方法一：卸载 GPU 驱动

卸载 GPU 驱动可以彻底禁用 GPU，但需要注意的是，这样会影响到其他需要使用 GPU 的进程。

# 查看已安装的 GPU 驱动
lspci -k | grep -EA2 'VGA|3D'
# 卸载 NVIDIA 驱动
sudo apt-get remove --purge nvidia*
# 卸载 AMD 驱动
sudo apt-get remove --purge fglrx*

方法二：禁用 GPU 核心

禁用 GPU 核心可以在不影响其他进程的情况下禁用 GPU。

# 禁用 NVIDIA 核心
sudo tee /etc/modprobe.d/blacklist-nvidia.conf <<< "blacklist nvidia"
sudo update-initramfs -u
# 禁用 AMD 核心
sudo tee /etc/modprobe.d/blacklist-radeon.conf <<< "blacklist radeon"
sudo update-initramfs -u

方法三：设定环境变量

设定环境变量可以在当前进程中禁用 GPU。

# 禁用 NVIDIA
export CUDA_VISIBLE_DEVICES=""
# 禁用 AMD
export AMD_VISIBLE_DEVICES=""

总结

以上是几种禁用 GPU 的方法，根据不同的需求选择合适的方法即可。当然，在使用之前需要确认一下你是否真的需要禁用 GPU。