一、安装nvidia显卡驱动，cuda，cudnn

sudo pacman -S nvidia cuda  cudnn

二、安装tensorflow-gpu

pip install tensorflow-gpu

三、问题补充

1. 若cuda版本太高11.1或者10.2,tensorflow-gpu2 目前最高只支持cuda10.1，用不了怎么办?

1）去官方仓库下载旧版本，链接如下：

https://archive.archlinux.org/packages/c/cuda/

下载好后进行安装(推荐用uget下载)

sudo pacman -U cuda-10.1xxxx.pkg.tar.xz

2)使用downgrade，进行降级

yay -S downgrade

运行下列命令:

downgrade cuda

选择需要降级的版本序号进行降级

2.若降级后防止cuda更新的时候升级了怎么办？

编辑pacman.conf，取消IgnorePkg前面的注释，并在=号后添加cuda,对cuda排除更新，具体操作如下：

sudo vim /etc/pacman.conf

修改后保存退出(:wq)

测试排除结果：

若出现上面这个即为排除成功，若想取消排除很简单，注释掉IgnorePkg即可

3）若pip下载tensorflow-gpu慢怎么办？

更换pip源,采用国内源（阿里，清华等），具体做法如下

在用户目录下创建.pip文件夹，在文件夹内创建并编辑pip.conf文件

mkdir ~/.pip
vim ~/.pip/pip.conf

内容填充如下:

[global] 
index-url = http://mirrors.aliyun.com/pypi/simple/ 
[install] 
trusted-host=mirrors.aliyun.com 

到此便结束了，若有疑问还有私信，若内容有误欢迎纠正