一、安装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
到此便结束了,若有疑问还有私信,若内容有误欢迎纠正