首页 > 其他分享> 文章详细

tensorflow物理机环境部署

2022-03-09 17:01:01 阅读：221 来源： 互联网

标签：x86 部署 64 cuda nvidia tensorflow root el7 物理

官网推荐跑tensorlow用docker环境，不过老板要求用物理机，咱就搞就是了

参考官网英文文档

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions

首先说下我的环境，centos7.8,1070ti显卡

1、

[root@node14 maintenance-item-match]# uname -r
3.10.0-1160.el7.x86_64

[root@node14 maintenance-item-match]# rpm -qa|grep kernel
kernel-3.10.0-1160.el7.x86_64
kernel-devel-3.10.0-1160.el7.x86_64
kernel-devel-3.10.0-1160.53.1.el7.x86_64
kernel-headers-3.10.0-1160.53.1.el7.x86_64
[root@node14 maintenance-item-match]#

如果这二个不一致，需要手动卸载安装，安装完重启

这边我就遇到了，因为环境不是干净的环境吧，导致我的内核需要降级到

3.10.0-1160.el7.x86_64

屏蔽默认带有的nouveau

[root@localhost 10:37:41 src]# vim /lib/modprobe.d/dist-blacklist.conf
blacklist nouveau
options nouveau modeset=0

[root@localhost 10:37:41 src]# mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
[root@localhost 10:37:41 src]# dracut /boot/initramfs-$(uname -r).img $(uname -r)

[root@localhost 10:37:41 src]# systemctl set-default multi-user.target

reboot

ls mod | grep nouveau 没有输出内容就行了

2、

https://developer.nvidia.com/cuda-11.2.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=rpmlocal

去下载安装文件

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-rhel7-11-2-local-11.2.0_460.27.04-1.x86_64.rpm

sudo rpm -i cuda-repo-rhel7-11-2-local-11.2.0_460.27.04-1.x86_64.rpmsudo yum clean all

sudo yum -y install nvidia-driver-latest-dkms cuda

sudo yum -y install cuda-drivers

3、添加环境变量

export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source /etc/profile

4检查

systemctl start nvidia-persistenced

systemctl enable nvidia-persistenced

nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Thu_Feb_10_18:23:41_PST_2022
Cuda compilation tools, release 11.6, V11.6.112
Build cuda_11.6.r11.6/compiler.30978841_0

标签：x86,部署,64,cuda,nvidia,tensorflow,root,el7,物理
来源： https://www.cnblogs.com/whitelittle/p/15985984.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

tensorflow物理机环境部署

屏蔽默认带有的nouveau