现象
突然心血来潮,要换Tensorflow2.0了本来打算换Python了,但看了看Tensorflow官网上的教程比较不错,又打算继续搞Tensorflow了。遇到一个问题就是已启动Tensorflow 2.0 GPU版本整个电脑被卡死,必须重新启动电脑。
我的电脑跑Tensorflow1.4 GPU版本一点问题也没有。CUDA 10.0版本,显卡驱动是410版本。完全符合官方标准:
解决方式
咋办呢?直接卡死了,连报错都看不到,我认为和驱动有很大的关系。换个驱动试试吧。我换成了430版本的。太走运了,一下子就成功了。看一下图片:
对于应该使用NVIDIA官网的驱动?还是使用ubuntu自带的驱动呢?我的经验是使用Ubuntu自身带的驱动。