我已经使用conda成功安装了tensorflow-gpu。当我测试执行导入张量流时,我遇到了上述问题。任何想法?我已经检查了驱动程序,正确安装了nvidia工具包和cudnn并分别设置了PATH,LD_LIBRARY_PATH和CUDA_HOME的值。
...星期五11月23日12:00:18
+ ------------------------------------------------- ----- +
| NVIDIA-SMI 340.107驱动程序版本:340.107 |
| ------------------------------- + ----------------- ----- + ---------------------- +
| GPU名称持久性-M |总线编号Disp.A |挥发性不佳。 ECC |
|风扇温度性能:用法/上限|内存使用| GPU实用计算M。
| ============================== + ================= ===== + ==================== |
| 0 Quadro FX 5600关闭| 0000:02:00.0 N / A | N / A |
| 61%77C P0不适用/不适用| 2MiB / 1535MiB | N / A默认|
+ ------------------------------- + ----------------- ----- + ---------------------- +
+ ---------------------------------------------- ------------------------------- + |计算过程:GPU内存| | GPU PID进程名称用法| | ================================================= ========================== | | 0不支持| + ------------------------------------------------- ---------------------------- +
答案 0 :(得分:0)
您需要使用cuDNN 3拥有适当的最低CUDA支持(似乎是7 https://askubuntu.com/questions/988787/nvidia-cuda-theano-could-not-find-symbol-cudeviceprimaryctxgetstate)。
如有可能,请升级驱动程序以获取该版本。
否则,请使用tensorflow-mkl
或用于较旧的CPU型号tensorflow-eigen
。
答案 1 :(得分:0)
我遇到了同样的问题-我已经在Ubuntu计算机上安装了必需的NVIDIA驱动程序,但是当我尝试导入Tensorflow时,我遇到了同样的错误。
所以对我来说,问题是我使用的是Nouveau驱动程序而不是NVIDIA驱动程序。为了解决此问题,您需要转到System Settings
-> Software & Updates
-> Additional Drivers
并选择选项Using NVIDIA binary driver ...
,然后单击Apply Changes
按钮。然后重新启动就可以了。