Ubuntu使用GPU深度学习加速的环境配置_ubuntu怎么使用电脑的gpu

1. 下载GPU驱动

去官网根据条件下载对应的GPU驱动文件，得到.run文件

2. 禁用Nouveau 驱动

禁用 Nouveau 驱动

3. 安装驱动（选择这一步，则在安装CUDA时可以不选择安装驱动；否则就需要选择安装CUDA时的驱动）

依次执行以下命令:

sudo apt install make
sudo apt install gcc
sudo chmod 777 xx.run
sudo apt-get remove --purge nvidia*
sudo bash xx.run
1
2
3
4
5

然后出现选择时，一路默认即可

最后验证是否安装成功，使用命令nvidia-smi，会出现以下证明成功了：

4. 安装CUDA

去官网下载相应版本的CUDA文件，文件格式为.run
在选择安装内容时，可以将驱动模块取消

sudo chmod 777 xx.run
sudo bash xx.run
1
2

安装完成后会出现如下信息：

意味着我们需要配置环境变量，配置环境变量相关教程参考：添加cuda到bashrc环境变量

安装成功后，使用命令nvcc -V查看CUDA版本：

5. 安装CUDNN

5.1 若是deb文件

官网下载对应版本的CUDNN，我下载的是deb版本，然后使用命令安装:sudo dpkg -i xxx.deb即可安装成功

• 注意：如果使用Paddle-GPU版本，可能会报错如下：
  
  这是因为cudnn虽然安装了， 但是没有指向，导致paddle找不到，那么参考这个教程：百度Paddlepaddle-GPU运行时出现Error: Cannot load cudnn shared library. Cannot invoke method cudnnGetVersion，只需要将libcudnn.so文件连接好即可，另一个so可不管