我想在Windows PC上同时支持Visual Studio 2019(需要CUDA 10.1)和TensorFlow-GPU 1.14(需要CUDA 10.0)。有什么方法吗?
我只安装了CUDA 10.0和CUDA 10.1,并将两个目录都添加到环境变量CUDA_PATH中。 cuDNN已经安装。
结果是Visual Studio可以检测到CUDA,但TensorFlow无法检测到。
答案 0 :(得分:2)
是的,系统上可以存在多个版本的CUDA工具包,并且可以由不同的应用程序使用。
您如何安装TensorFlow-GPU?如果要自己编译,则可以在配置期间指定要使用的CUDA版本的路径。如果要安装一组预构建的二进制文件(例如,使用诸如Anaconda之类的二进制文件),则说明已针对特定版本的CUDA工具包进行了构建;您需要获取为所需的CUDA工具包编译的二进制文件的不同版本,或者自己构建。
如果您使用Anaconda安装TensorFlow-GPU,则还应该获得运行已安装的TensorFlow-GPU所需的正确版本的CUDA工具包。它会为您解决这些依赖性。