Tensorflow 2.10是最后一个在本地windows上支持GPU的版本。从2.11版本开始,需要在windows WLS2(适用于 Linux 的 Windows 子系统)上安装才能使用GPU。
在anaconda shell控制台中,切换至虚拟环境, 安装TensorFlow
这是用的豆瓣镜像,要装不同版本只需修改版本号
pip install tensorflow_gpu==2.10.0 -i https://pypi.douban.com/simple --trusted-host pypi.douban.com
其他指令
# 卸载普通版本的tensorflow
pip uninstall tensorflow
# 卸载intel版本的tensorflow
pip uninstall tensorflow-intel
无镜像pip安装 速度超慢 但是会自动安装兼容当前python版本的最新版本
pip install --upgrade pip
pip install tensorflow
测试代码
import tensorflow as tf
import keras
print(tf.__version__)
print(keras.__version__)