美文网首页
CentOS 7.4 yum 安装 cuda 各种踩坑指南

CentOS 7.4 yum 安装 cuda 各种踩坑指南

作者: 9efe1db2c646 | 来源:发表于2019-01-15 16:48 被阅读34次

被cuda折磨得死去活来,下面是经验总结:

  • 原则一:GPU和CUDA分开安装
  • 原则二:GPU驱动包保持最新,CDUA按需安装

一、下载最新GPU驱动并安装

下载最新驱动(解决内核不兼容问题)及需要的CUDA安装包
官网下载最新驱动
官网下载cuda 9.0

二、安装必要的基础库(可参考文末的yum加速配置)

yum install gcc 
yum install kernel*
yum install epel-release
yum install --enablerepo=epel dkms

三、屏蔽默认带有的nouveau

3.1 使用su命令修改文件

sudo vi /lib/modprobe.d/dist-blacklist.conf

修改内容如下:

#将nvidiafb注释掉。
#blacklist nvidiafb

...

#然后添加以下语句:
blacklist nouveau
options nouveau modeset=0

3.2 重建initramfs image

mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut /boot/initramfs-$(uname -r).img $(uname -r)

3.3 reboot 重启

3.4(可选) 查看nouveau是否已经禁用

lsmod | grep nouveau

三、安装驱动

chmod 777 显卡驱动.run
sh 显卡驱动.run

chmod 777 CUDA驱动.run
sh CUDA驱动.run

四、下载加速配置

4.0 禁用ipv6

解决yum无法解析

vi /etc/default/grub

找到GRUB_CMDLINE_LINUX这一行,在双引号内加入如下内容,注意与其他项目之间使用空格隔开:

GRUB_CMDLINE_LINUX="ipv6.disable=1"

在终端输入如下内容:

grub2-mkconfig -o /boot/grub2/grub.cfg
reboot

重启完成之后,使用以下命令查看IPv6模块是否已经全部关闭(截图为未禁用IPv6情况,禁用之后是没有内容的):

lsmod | grep ipv6

4.1 配置yum阿里源

4.1.1 禁用ipv6

在CentOS 7下不要直接修改/etc/resolv.conf,而是应该使用Systemctl restart NetworkManager命令,它会根据/etc/sysconfig/network-scripts/ifcfg-eth0自动生成etc/resolv.conf。

#第一步:备份本地yum源
yum install wget
mv /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.backup

#第二步:下载阿里yum源文件
wget -O /etc/yum.repos.d/CentOS-Base.repo http://mirrors.aliyun.com/repo/Centos-7.repo

#第三步:生成yum缓存
yum clean all
yum makecache

4.2 配置pip 阿里源


Linux系统:

mkdir ~/.pip

cat > ~/.pip/pip.conf << EOF

[global]
trusted-host=mirrors.aliyun.com
index-url=https://mirrors.aliyun.com/pypi/simple/
EOF

4.3 配置conda阿里源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

4.4 安装Anaconda3

官网下载sh安装文件并安装
激活

echo 'export PATH="~/anaconda3/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc

yum install -y python-qt4
conda create -n lutz python=3.6.5 -y
source activate lutz
conda install tensorflow-gpu -y
pip install scikit-image
pip install keras==2.0.8
pip install opencv-python==3.4.5.20
conda install nb_conda

nohup tensorboard --host 0.0.0.0 --logdir=./logs > tensorboard .log &

nohup jupyter notebook --ip 0.0.0.0 --no-browser --allow-root > jupyter.log &

安装jupyter代码提示

conda install jupyter
pip install jupyter_contrib_nbextensions
jupyter contrib nbextension install --user
pip install jupyter_nbextensions_configurator
jupyter nbextensions_configurator enable --user
nbextensions勾选Hinterland

相关文章

网友评论

      本文标题:CentOS 7.4 yum 安装 cuda 各种踩坑指南

      本文链接:https://www.haomeiwen.com/subject/dhmedqtx.html