Centos7 安装Anaconda和GPU版本TensorFlow

本文介绍了在CentOS7上安装GPU驱动、CUDA、cuDNN、Anaconda和TensorFlow的详细步骤。首先,禁用nouveau驱动并安装NVIDIA驱动,然后按照CUDA和cuDNN官方手册安装相应组件。接着,下载Anaconda并设置环境变量,最后创建TensorFlow环境并安装GPU支持的TensorFlow。
摘要由CSDN通过智能技术生成

写在前面,在安装之前要先考虑版本对应的问题

如何查看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和驱动的对应关系有篇博客讲的很细:(待补充)

一. 准备

  1. 禁用 nouveau 驱动
lsmod | grep nouveau

查询是否预安装了nouveaud。若无,跳过以下2-4步骤。

  1. 编辑 vi /etc/modprobe.d/blacklist.conf ,在文件后面加入blacklist nouveau

  2. 备份 the initramfs file

mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
  1. 重新建立 the initramfs file
dracut -v /boot/initramfs-$(uname -r).img $(uname -r) --force
  1. 安装依赖
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值