debian安装tensorflow-GPU

一、安装anaconda

文章

二、安装CUDA

1、查看linux内核版本

lsb_release -a

如果说没有该命令，可以apt-get install 一下

2、确认GPU版本

lspci | grep -i nvidia

GPU要求算力3.0以上，可以对照型号通过下面链接查看自己显卡是否支持链接

3、确认是否安装gcc

gcc --version

4、禁用nouveau

• 查看相关模块文件。如果没有该模块，可能是没有配置sbin环境。export PATH=$PATH/sbin

modinfo  -F filename nouveau

• 使用blacklist禁用

sudo nano /etc/modprobe.d/blacklist-nouveau.conf
#在文件中加入以下两行
blacklist nouveau
options nouveau modeset=0

• 更新一下

update -initramfs -u

• 重启
• 输入lsmod | grep nouveau 如果没有输出就说明禁用了

5、安装NAVIDIA驱动

• 去官网找与自己相匹配的驱动链接
• bash 文件名.sh运行，如果有什么存在X服务器的报错我们就要进入文本模式安装。
• 在终端中输入，systemctl set-default multi-user.target，表示下次启动进入文本模式
• 重启，进入文本模式后，我们进入到下载文件所在目录。bash 文件名.sh运行
• 跟着提示就可以安装成功了
• 输入systemctl set-default graphical.target,表示下次启动进入图形界面
• 重启，在终端中输入nvidia-smi，如果有输出就成功了

6、安装CUDA11.6

• 选择CUDA与cudnn匹配版本。链接
• 下载CUDA，根据自己的版本下载链接
• 在安装过程中会有点小问题，注意看提示。（我这里选的都是yes）
• 配置环境，我的默认安装位置是在/usr/local下的。

export LD_LIBRARY_PATH=/usr/local/cuda/lib

export PATH=$PATH:/usr/local/cuda/bin

• 输入nvcc -V如果有版本输出，安装成功

7、安装cudnn（需要注册才能下载）

• 下载对应版本的cudnn链接
• 解压，解压后文件中有include和lib文件夹
• 将include文件夹中所有文件复制到/usr/local/cuda/include中
• 将lib文件夹中所有文件复制到/usr/local/cuda/lib64中
• cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A2如果有输出就是安装成功

8、安装tensorflow-gpu

pip install tensorflow-gpu -i htttps://pypi.douban.com/simple/