TensorFlow-in-nvidia-docker-fail

2018-05-30  本文已影响0人  burning_yang

(转载请注明作者和出处:https://yangningbocn.github.io 未经允许请勿用于商业用途)

背景:
Ubuntu
TensorFlow


问题描述:
出现错误:
E TensorFlow in nvidia-docker: failed call to cuInit: CUDA_ERROR_UNKNOWN
然后就用CPU在跑


解决方法:

sudo update-alternatives --config x86_64-linux-gnu_gl_conf
解决之前

选择 /usr/lib/x86_64-linux-gnu/mesa/ld.so.conf
就好了
作为一个强迫症,你以为这样就完了吗,不!
我要把其他三个删掉
这个命令是用管理员的权限打开文件浏览器,然后删除那些不用的三个选项。这三个选项是我之前安装NVIDIA驱动后并卸载的残留

sudo nautilus 
解决之后的图片

在/usr/lib/下 ctrl+f 输入nvidia 然后就能找到那些不用的选项删除就OK了。

上一篇下一篇

猜你喜欢

热点阅读