时至2019年12月21日,tensorflow2.0依然只能在cuda10.0及以下环境运行。  

如果英伟达驱动版本很高,并自带cuda10.2之类,需要自己指定安装在目录中的cuda10和cnn。

tensorflow GPU 版 cuda10.0安装 注意点
 

比如在anaconda环境中,promot输入以上指令。

 

 

此外如需编译 pip 软件包需要其他依赖,见以下链接。

https://www.tensorflow.org/install/source_windows#build_the_pip_package

相关文章: