本人笔记本是Thinkpad T470系列,默认显卡是Geforce 940MX,驱动程序默认支持CUDA9.0,而最新的TensorFlow 2.2需要支持CUDA10以上,具体环境安装如下:
1.Python和pip安装省略(请自行搜索安装),建议Python 3.7.*以上
2.升级最新的NVIDIA显卡驱动,要到thinkpad官网下载,我下载最新的支持CUDA10.1。
3.https://developer.nvidia.com/cuda-toolkit-archive,下载最新的CUDA Toolkit
4.https://developer.nvidia.com/rdp/cudnn-download?spm=a2c4e.10696291.0.0.1df819a4HJWSTe,下载最新的cuDNN( is a GPU-accelerated library of primitives for deep neural networks)
5.使用#>pip install tensorflow 安装最新版的tensorflow
6.命令行模式敲入,#>python,进入shell模式,
#>import tensorflow as tf
#>a=tf.constant(888)
#>a
7.如果没有报错,既可以使用tensorflow。
FAQ:
1.不要从nvidia官方网站下载显卡驱动,会报错
2.如果上述都报错,可以使用降级安装处理,比如pip install tensorflow==1.9.0