升级到带有一些cuda问题的tensorflow2.0 alpha

时间:2019-03-24 08:23:39

标签: tensorflow cuda

我已经在Windows计算机上使用tensorflow1.12和Cuda9.0。 今天,我想将tensorflow升级到2.0 Alpha,因此我通过Anaconda创建了一个新环境并pip install tensorflow-gpu 2.0 alpha。据我所知,当我不想卸载现有的9.0时需要cuda 10.0,所以我尝试了conda install cudatoolkit cudnn。结果显示cudatoolkit:10.0.130-0和cudnn:7.3.1-cuda10.0_0都自动下载。

然后,我尝试tensorflow2.0。 import tensorflow不会引发错误。 但是,当我使用x=tf.constant([1])时,它会引发这种情况

cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

我认为问题是我的tensorflow2.0找不到下载的cudatoolkit 10.0,尽管它们在同一anaconda环境中。那我该怎么解决

1 个答案:

答案 0 :(得分:1)

将我的GPU驱动程序版本升级到最新版本可以轻松解决。