window11 TensorFlow GPU efficientnet图像分类

环境要求:cuda11.2 cuDNN8.1(根据自己环境安装对应tensorflow版本)
官网:https://tensorflow.google.cn/install/pip?hl=zh-cn

1、安装Anaconda
2、创建一个新的conda环境并激活

conda create -n tf37 python=3.7
conda activate tf37

3、安装tensorflow-gpu

conda install tensorflow-gpu=2.6

4、安装efficientnet

pip install efficientnet

验证tensorflow安装及GPU支持

import tensorflow as tf
print(tf.test.is_built_with_cuda())
#Ture
 
print(tf.config.list_physical_devices('GPU'))
#[]
 
print(tf.test.is_gpu_available())
#False
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容