centos7配置tensorflow gpu環境

2021-10-09 08:51:52 字數 1966 閱讀 3055

凡對本文有任何疑惑可加qq**流：1081332609

作業系統：centos7.x

顯示卡型號：geforce gtx 750 ti

python版本：python3.6

tensorflow版本：tensorflow-gpu==1.14.0

nvidia驅動：nvidia-linux-x86_64-440.36.run

cuda：cuda_10.0.130_410.48_linux.run

cudnn: cudnn-10.0-linux-x64-v7.6.4.38.tgz

會顯示顯示卡的型號

lspci | grep -i nvidia
gcc --version
其中$(uname -r)就是個引數，uname -r表示kernel的版本號。

sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
systemctl stop gdm.service
開啟如下檔案

vi /usr/lib/modprobe.d/dist-blacklist.conf
編輯，並在最後新增如下兩行,並儲存

blacklist nouveau


options nouveau modeset=0
備份 initramfs 檔案

sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
重建 initramfs 檔案

dracut /boot/initramfs-$(uname -r).img $(uname -r)
重啟電腦

reboot

sudo sh nvidia-linux-x86_64-440.36.run --kernel-source-path=/usr/src/kernels/$(uname -r)
在accept的頁面選擇accept，

在32-bit頁面選擇no

在x- configuration頁面選擇yes

sudo sh cuda_10.0.130_410.48_linux.run
編輯bashrc檔案

sudo vi ~/.bashrc
在最後新增如下內容，並儲存（注意cuda的版本號）

export cuda_home=/usr/local/cuda-10.0


export path=/usr/local/cuda-10.0/bin:$path


export ld_library_path=/usr/local/cuda-10.0/lib64:$ld_library_path


export ld_library_path="/usr/local/cuda-10.0/lib:${ld_library_path
tar -xvzf cudnn-10.0-linux-x64-v7.6.4.38.tgz 


cp cuda/include/* /usr/local/cuda/include 


cp cuda/lib64/* /usr/local/cuda/lib64
建議採用國內映象加速安裝

pip install -i  --trusted-host mirrors.aliyun.com tensorflow-gpu==1.14.0
八年抗戰，終於取得最後勝利！！！

CentOS7基礎配置

預設centos7安裝將選擇最小安裝，完後是沒有任何配置的，需要手動配置需要的各種工具，同時dhcp也是出於關閉狀態，需要手動開啟。輸入命令vi etc sysconfig network scripts ifcfg eno16777736 type ethernet bootproto dhcp ...

CentOS 7 基礎配置

3 準備jdk和hadoop安裝包 centos7的靜態ip設定與以前版本的linux有細微的差別，這也是需要注意的地方。作者 為了配置 有效的靜態ip，嘗試了很多的方法。選擇虛擬機器的編輯選項 虛擬網路編輯器 管理員模式開啟 在上圖中已經標註出需要注意的地方，同時需要檢視nat設定 設定閘道器ip...

CentOS 7 配置網路

檢視ip位址 ip addr結果如下，可以看看網絡卡名稱 ens33 1 lo mtu 65536 qdisc noqueue state unknown qlen 1 link loopback 00 00 00 00 00 00 brd 00 00 00 00 00 00 inet 127.0....