我想使用启用的tensorflow版本。我读到它最适合CUDA 7.5,但是我使用的服务器已安装了CUDA版本5.5。
我可以用CUDA 5.5配置TensorFlow吗?如果是,怎么样?我已经在虚拟环境中安装了TensorFlow。
来自安装文档:
为了使用GPU支持构建或运行TensorFlow,NVIDIA的CUDA工具包(> = 7.0)和CUDNN(> = V2)都需要安装。
如果您不从源安装,则需要使用7.5版。
还确保您具有兼容的GPU:
TensorFlow GPU支持需要具有NVIDIA COMPUTE功能的GPU卡> = 3.0。受支持的卡包括但不限于:
- Nvidia Titan
- Nvidia Titan X
- nvidia k20
- nvidia k40