在Windows10安装TensorFlow 2.2.0 GPU版

TensorFlow GPU v2.2.0版本的具体安装步骤如下:
第一步,在Anaconda中创建一个Python=3.6的tf_gpu虚拟环境

创建tf_gpu虚拟环境

第二步,输入命令:pip install --ignore-installed --upgrade tensorflow==2.2.0 安装TensorFlow V2.2.0 GPU版
第三步,输入测试命令:python -c "import tensorflow as tf;print(tf.math.add(1,2))",此时会收到错误信息:dlerror: cudart64_101.dll not found
收到错误信息

第五步,输入命令:conda install cudatoolkit=10.1 cudnn=7.6.5 安装CUDA Toolkit v10.1和cuDNN 7.6.5
第六步,输入测试命令:python -c "import tensorflow as tf;print(tf.math.add(1,2))",此时会收到信息Successfully opened dynamic library cudart64_101.dll,说明CUDA Toolkit v10.1和cuDNN 7.6.5安装成功
第七步,输入测试代码:

import tensorflow as tf
hello = tf.constant('hello,tensorflow!')
print(hello.device)
print(hello)

若获得下图所示结果,说明:TensorFlow 2.2.0 GPU版安装成功!
TensorFlow 2.2.0 GPU版安装成功

安装Keras:
首先检查Keras与TensorFlow的对应关系:https://docs.floydhub.com/guides/environments/
TensorFlow 2.2.0对应Keras 2.3.1,但实践下来2.4.3配置最好,建议安装2.4.3
用命令:

pip install keras==2.4.3

安装keras

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容