使用 TF2.1 后 對應(yīng)的 Cuda 和舊版本的驅(qū)動(dòng)不支持了,因此需要升級一下驅(qū)動(dòng)溅潜。
檢查最新驅(qū)動(dòng)
首先在 NVIDIA 的網(wǎng)站里查找驅(qū)動(dòng)的最新版本,下載對應(yīng)的 .run
文件即可。
卸載原有驅(qū)動(dòng)
使用 chmod a+x
給下載的.run
文件添加執(zhí)行權(quán)限芜赌,然后使用 --uninstall
參數(shù)進(jìn)行舊版驅(qū)動(dòng)卸載:
chmod a+x NVIDIA-Linux-x86_64-418.126.02.run
./NVIDIA-Linux-x86_64-418.126.02.run --uninstall
安裝新驅(qū)動(dòng)
./NVIDIA-Linux-x86_64-418.126.02.run
如無意外很快就能安裝完成其爵。
更新 GCC 7.X.X
首先安裝 devtoolset 包冒冬,-7
對應(yīng)7.X.X版本。
yum install centos-release-scl
yum install devtoolset-7
執(zhí)行激活 gcc 版本(該終端內(nèi)有效)摩渺,使其在 zsh 中生效(或bash):
scl enable devtoolset-7 zsh
或直接寫入 ~/.zshrc 中简烤。
source /opt/rh/devtoolset-7/enable