美文网首页
window11 TensorFlow GPU efficien

window11 TensorFlow GPU efficien

作者: wcdd | 来源:发表于2024-03-18 21:29 被阅读0次

环境要求: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

相关文章

网友评论

      本文标题:window11 TensorFlow GPU efficien

      本文链接:https://www.haomeiwen.com/subject/hfyzzdtx.html