TensorFlow - 在 Mac 上安装 GPU

3

运行python3脚本后,我得到了以下三个语句,但不知道这三个错误来自哪里。我正在使用cudnn v5.0,但显然我在安装过程中出了些问题。希望能得到帮助。

I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcublas.dylib locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcudnn.dylib locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcufft.dylib locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcuda.1.dylib locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcurand.dylib locally
number of elements at final reshape = %d. 61440
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:892] OS X does not support NUMA - returning NUMA node zero
I tensorflow/core/common_runtime/gpu/gpu_init.cc:102] Found device 0 with properties:
name: GeForce GT 750M
major: 3 minor: 0 memoryClockRate (GHz) 0.9255
pciBusID 0000:01:00.0
Total memory: 2.00GiB
Free memory: 1.28GiB
I tensorflow/core/common_runtime/gpu/gpu_init.cc:126] DMA: 0
I tensorflow/core/common_runtime/gpu/gpu_init.cc:136] 0:   Y
I tensorflow/core/common_runtime/gpu/gpu_device.cc:838] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce GT 750M, pci bus id: 0000:01:00.0)
E tensorflow/stream_executor/cuda/cuda_dnn.cc:354] could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED
E tensorflow/stream_executor/cuda/cuda_dnn.cc:361] error retrieving driver version: Invalid argument: expected %d.%d or %d.%d.%d form for driver version; got ""
E tensorflow/stream_executor/cuda/cuda_dnn.cc:321] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM
F tensorflow/core/kernels/conv_ops.cc:457] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms)
Abort trap: 6

你可以尝试禁用SIP(系统完整性保护)。 - Yaroslav Bulatov
那有帮助吗? - gbachik
我没有尝试过,你确定禁用SIP是个好主意吗?仅仅从一点点了解它的情况来看。 - mattdns
当我安装我的TensorFlow GPU时,禁用SIP是必须的(即,当我的SIP恢复默认设置时,TF GPU停止工作。我再次关闭它,它又开始工作了)。 - Yaroslav Bulatov
1个回答

1

你可能安装了错误的cuDNN!

我曾经也遇到过同样的问题,通过替换兼容性更好的cuDNN版本解决了我的问题:

enter image description here


我想我现在安装了多个cudnn版本。卸载其他版本的最佳方法是什么?我只需要删除头文件和库文件吗? - mattdns
1
没错!只需删除您在安装步骤中添加的文件,您就可以继续了。 - gbachik

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接