环境要求: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
网友评论