标签：Linux 安装 nvidia Toolkit CUDA NVIDIA 分割线

缘起std::bad_alloc

程序跑到中途抛出异常：

按照网上说法，将内存从8G扩大到了32G，无效。仍然抛出该异常。

查看TensorFlow是gpu版本还是cpu版本

from tensorflow.python.client import device_lib 
print(device_lib.list_local_devices())

发现没用上GPU，一直是CPU在跑。

检查驱动问题

验证显卡驱动是否装好：

nvidia-smi

抛出错误：

检查驱动是否安装：

nvcc -V

抛出错误：

按照提示指令安装toolkit后：

发现驱动存在。

尝试解决nvidia-smi的错误：

首先找到NVIDIA驱动版本：

ls /usr/src | grep nvidia

发现两个驱动的版本有问题。

尝试输入指令：

sudo apt install dkms
sudo dkms install -m nvidia -v 450.80.02

---

分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线分割线

---

安装CUDA

1 检查系统是否有支持 CUDA 编程的 GPU

查看当前系统的 GPU 型号：

lspci | grep -i nvidia

可以直接安装 CUDA Toolkit，系统将自动安装与其版本匹配的 NVIDIA Driver

2 安装 CUDA Toolkit

在安装 CUDA Toolkit 前，要确保系统安装了 gcc 和 make。如果希望使用 C++ 进行 CUDA 编程，需要安装 g++。如果想要运行 CUDA 例程序，需要安装相应的依赖库。

sudo apt update # 更新 apt
sudo apt install gcc g++ make # 安装 gcc g++ make
sudo apt install libglu1-mesa libxi-dev libxmu-dev libglu1-mesa-dev freeglut3-dev # 安装依赖库

在 CUDA Toolkit 的下载页面选择系统版本和安装方式，下载并运行 runfile。

下载 CUDA Toolkit

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run

安装 CUDA Toolkit

sudo sh cuda_11.1.0_455.23.05_linux.run

安装失败，返回代码是256。log如下：

参考文献

1 无法连接NVIDIA驱动：NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver
2 Linux NVIDIA显卡驱动安装
3 nvidia-smi 报错：无法与 nvidia driver 通信
4* Linux 下的 CUDA 安装和使用指南

标签：Linux,安装,nvidia,Toolkit,CUDA,NVIDIA,分割线
来源： https://www.cnblogs.com/gangao/p/16648005.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。