我已经在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环境中。那我该怎么解决
答案 0 :(得分:1)
将我的GPU驱动程序版本升级到最新版本可以轻松解决。