写在前面,在安装之前要先考虑版本对应的问题
如何查看GPU/CUDA/CUDNN/Python/Tensorflow/Anaconda的对应版本呢?
我使用的GPU是P4000,驱动安装的是NVIDIA-Linux-x86_64-450.57.run,配置了CUDA10.1,cudnn-10.1-linux-x64,Python3.8.3,Anaconda最新版,tensorflow2.2.0
本人初学,网上有一些对应的资料可以查,但是新版的不太好找到,所以我是先装了anaconda和tensorflow,之后根据tensorflow的warning提示安装的cuda和cudnn。但是我下面的安装顺序还是正常的,其实只要定义环境变量之后就可以用gpu版本的tensorflow了。
安装GPU驱动
参考博客:https://www.cnblogs.com/YSPXIZHEN/p/11466145.html
GPU和驱动的对应关系有篇博客讲的很细:(待补充)
一. 准备
- 禁用 nouveau 驱动
lsmod | grep nouveau
查询是否预安装了nouveaud。若无,跳过以下2-4步骤。
-
编辑
vi /etc/modprobe.d/blacklist.conf
,在文件后面加入blacklist nouveau
-
备份 the initramfs file
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
- 重新建立 the initramfs file
dracut -v /boot/initramfs-$(uname -r).img $(uname -r) --force
- 安装依赖