TensorFlow 2.10 是最后一个支持在原生Windows上使用GPU的TensorFlow版本

注意:TensorFlow 2.10 是最后一个支持在原生Windows上使用GPU的TensorFlow版本。从TensorFlow 2.11开始,您需要在WSL2中安装TensorFlow,或者安装tensorflow或tensorflow-cpu,并且可选地尝试TensorFlow-DirectML插件

bash 复制代码
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
# Anything above 2.10 is not supported on the GPU on Windows Native
python -m pip install "tensorflow<2.11"
# Verify the installation:
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

pip install tensorflow==2.10 -i https://pypi.tuna.tsinghua.edu.cn/simple

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0

注意用这个命令的话cuda版本大于等于11.2

conda install cudatoolkit=11.2 cudnn=8.1.0

conda install cudatoolkit=11.3 cudnn=8.2.1

python 复制代码
import tensorflow as tf
version=tf.__version__  #输出tensorflow版本
gpu_ok=tf.test.is_gpu_available()  #输出gpu可否使用(True/False)
print("tf version:",version,"\nuse GPU:",gpu_ok)
相关推荐
love530love几秒前
【02】EPGF 架构搭建教程之 Python 多版本配置
运维·开发语言·人工智能·windows·python·架构·epgf 架构
山烛1 分钟前
OpenCV :基于 Lucas-Kanade 算法的视频光流估计实现
人工智能·opencv·计算机视觉·音视频·图像识别·特征提取·光流估计
小白学C++.3 分钟前
大模型agent综述:A Survey on Large Language Model based Autonomous Agents
人工智能·语言模型·自然语言处理
CVer_4 分钟前
图像处理:simd快速灰度化
图像处理·人工智能·计算机视觉
机器之心10 分钟前
Sam Altman发文,透露OpenAI正在干的大事业
人工智能·openai
0xCode 小新15 分钟前
鸿蒙ArkTS Canvas实战:转盘抽奖程序开发教程(基础到进阶)
人工智能·深度学习·机器学习·华为·ai·harmonyos·canvas组件
水龙吟啸16 分钟前
从零开始搭建深度学习大厦系列-4.Transformer生成式大语言模型
人工智能·深度学习·自然语言处理·大语言模型
KubeSphere 云原生20 分钟前
云原生周刊:MetalBear 融资、Chaos Mesh 漏洞、Dapr 1.16 与 AI 平台新趋势
人工智能·云原生
黎燃22 分钟前
迈向星际智能:AI 自主决策如何重塑未来太空探索
人工智能
路遥芝麻27 分钟前
如何使用升腾C92主机搭建本地Linux编译服务器并通过Windows映射访问共享目录
linux·服务器·windows