Ubuntu安装GPU环境解决nVidia-smi无法使用的问题(亲测可用)
1.如果已有环境的话,删除命令如下所示:
sudo apt-get --purge remove nvidia*
sudo apt autoremove
To remove CUDA Toolkit:
sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt-get --purge remove "*nvidia*"
2.开始安装
1.curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
2.sudo apt-get update # 更新源
3.sudo apt-get install -y nvidia-container-toolkit 这个工具也是需要安装的,如果后面的命令执行出问题的话,这个命令执行完就需要重启电脑
4.sudo apt-get install dkms
5.安装nvidia-smi
ubuntu-drivers devices # 查看推荐安装的driver,安装推荐的driver,如下图所示
安装命令为:
sudo apt install nvidia-driver-545-open # 进行安装之后需要重启计算机
接下来输入
nvidia-smi即可查看对应的GPU使用情况
nvidia-smi即可查看对应的GPU使用情况
![image-20240620093751612](https://img-blog.csdnimg.cn/img_convert/ee9b8da11e01fcf459c7931f52548362.png)