一、安装内容
只有安装了硬件驱动和工具包才可以使用GPU
1. 硬件驱动
- 安装显卡驱动
- 安装Cuda
- 安装cuDNN
2. 工具包
- TensorFlow-gpu
- TensorFlow
- keras
- ······
二、安装方法
1.安装方法和过程
具体的安装过程：可以参考这篇blog写的很详细。
2.python和TensorFlow-GPU版本问题
CSDN的博客地址，有可能版本更新，或者其他原因，给出官网的地址官网地址，官网给出了很多包括Windows以及Linux\MACOS等。
2.cuda的安装
安装CUDA的时候同样需要查看版本，具体的CUDA可以查看官网,不过cuda会给出安装的指令，因此问题不会太大。
3.cudnn的安装
cudnn需要根据需求下载三个deb文件并安装，工具包图如下：
安装后需要使用下面的指令去安装上述的三个工具包
sudo dpkg -i xxxx.deb
在进行编译和测试的时候，会出现的两个bug：
- 无法创建test.c的文件
rm -rf *o
rm -rf mnistCUDNN
Linking agains cublasLt = false
CUDA VERSION: 10000
TARGET ARCH: x86_64
HOST_ARCH: x86_64
TARGET OS: linux
SMS: 30 35 50 53 60 61 62 70 72 75
/bin/sh: 1: cannot create test.c: Permission denied
/bin/sh: 1: cannot create test.c: Permission denied
g++: error: test.c: No such file or directory
g++: warning: ‘-x c’ after last input file has no effect
g++: fatal error: no input files
compilation terminated.
解决办法
解决这个有两个方法
1. 使用sudo chmod 777 [filename] 提高权限，ubantu一定要加sudo
2. 使用chattr -ia [filename] 去掉-ia的属性
- 出现没有FreeImage.h文件的bug
mnistCUDNN sudo make
CUDA_VERSION is 11010
Linking agains cublasLt = true
CUDA VERSION: 11010
TARGET ARCH: x86_64
HOST_ARCH: x86_64
TARGET OS: linux
SMS: 35 50 53 60 61 62 70 72 75 80 86
test.c:1:10: fatal error: FreeImage.h: No such file or directory
1 | #include "FreeImage.h"
| ^~~~~~~~~~~~~
compilation terminated.
解决办法
使用下面的指令去安装
sudo apt-get install libfreeimage3 libfreeimage-dev
三、总结
根据上述的内容，基本可以安装成功，唯一需要注意的就是版本不能出现问题，否则就清空重新开始。遇到其他bug可以直接搜索错误。
