使用 TF2.1 后 对应的 Cuda 和旧版本的驱动不支持了,因此需要升级一下驱动。
检查最新驱动
首先在 NVIDIA 的网站里查找驱动的最新版本,下载对应的 .run
文件即可。
卸载原有驱动
使用 chmod a+x
给下载的.run
文件添加执行权限,然后使用 --uninstall
参数进行旧版驱动卸载:
chmod a+x NVIDIA-Linux-x86_64-418.126.02.run
./NVIDIA-Linux-x86_64-418.126.02.run --uninstall
安装新驱动
./NVIDIA-Linux-x86_64-418.126.02.run
如无意外很快就能安装完成。
更新 GCC 7.X.X
首先安装 devtoolset 包,-7
对应7.X.X版本。
yum install centos-release-scl
yum install devtoolset-7
执行激活 gcc 版本(该终端内有效),使其在 zsh 中生效(或bash):
scl enable devtoolset-7 zsh
或直接写入 ~/.zshrc 中。
source /opt/rh/devtoolset-7/enable
网友评论