参考https://www.jianshu.com/p/1710d9b58a50
增加新yum源
yum install epel-release
安装内核
yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
安装GCC9
yum install centos-release-scl scl-utils-build -y &&
yum install devtoolset-9-toolchain -y &&
scl enable devtoolset-9 bash&& echo “source /opt/rh/devtoolset-9/enable” >>/etc/profile
#yum list all --enablerepo=‘centos-sclo-rh’ | grep “devtoolset-”
安装lspci工具
yum install pciutils #lspci | grep -i nvidia
#安装dkms(前面有新源,可以不手动安装)
#yum install dkms
下载cuda
wget https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda-repo-rhel7-11-2-local-11.2.2_460.32.03-1.x86_64.rpm
安装cuda
sudo rpm -i cuda-repo-rhel7-11-2-local-11.2.2_460.32.03-1.x86_64.rpm
安装
sudo yum clean all
sudo yum -y install nvidia-driver-latest-dkms cuda
CentOS系统中安装CUDA和Tensorflow的详细步骤

该文章提供了一步步在CentOS上安装CUDA和Tensorflow的指南,包括更新yum源,安装内核、GCC9,启用devtoolset-9,安装dkms,下载并安装CUDA和NVIDIA驱动,设置环境变量,以及解决可能遇到的问题,如NUMA节点错误。同时强调了Python版本和Tensorflow版本需与CUDA兼容。
最低0.47元/天 解锁文章
1008

被折叠的 条评论
为什么被折叠?



