> 文章列表 > 最精简:windows环境安装tensorflow-gpu-2.10.1

最精简:windows环境安装tensorflow-gpu-2.10.1

最精简:windows环境安装tensorflow-gpu-2.10.1

Tensorflow 2.10是最后一个在本地windows上支持GPU的版本

  • 1. 通过.whl文件方式安装
  • 2.创建anaconda虚拟环境
  • 3.安装对应的cuda与cudnn版本,local不必装cuda和cudnn
  • 4. 测试tensorflow gpu是否可用

1. 通过.whl文件方式安装

.whl文件的下载地址:
tensorflow-gpu: http://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/
tensorflow-cpu: https://pypi.org/project/tensorflow/#files

下载: tensorflow_gpu-2.10.1-cp310-cp310-win_amd64.whl

Note: 从2.11版本开始,需要在windows WLS2(适用于 Linux 的 Windows 子系统)上安装才能使用GPU。所以要在native-windows上使用GPU,就只能安装2.10.0版本及以下的版本,或者安装老版的tensorflow-gpu。
windows native: link, 将官网的语言调到 英文 模式才能看到
最精简:windows环境安装tensorflow-gpu-2.10.1

2.创建anaconda虚拟环境

(一)创建环境

conda create -n tf2.10 python=3.10

(二)激活 tf2 环境

conda activate tf2.10

(三)通过 pip 安装下载的 tensorflow_gpu-2.10.1-cp310-cp310-win_amd64.whl

pip install your-path/tensorflow_gpu-2.10.1-cp310-cp310-win_amd64.whl

最精简:windows环境安装tensorflow-gpu-2.10.1

tensorflow 2.10.1安装完毕, 并且安装了对应的:keras-2.10.0
最精简:windows环境安装tensorflow-gpu-2.10.1

3.安装对应的cuda与cudnn版本,local不必装cuda和cudnn

conda install cudatoolkit cudnn

Note: 可以为cudatoolkit cudnn指定版本,eg:cudatoolkit=11.3 cudnn=8.3
.最精简:windows环境安装tensorflow-gpu-2.10.1

4. 测试tensorflow gpu是否可用

import tensorflow as tf
tf.__version__tf.test.is_gpu_available()
tf.config.list_physical_devices('GPU')

最精简:windows环境安装tensorflow-gpu-2.10.1
至此,tensorflow-gpu 2.10.1 安装成功,愉快的玩耍吧。

.
Reference:
解决CUDA 11.6版本对应的tensorflow-gpu版本问题
2023年最新Windows安装GPU版本的tensorflow(含bug记录及解决)
Tensorflow找不到GPU:tensorflow 2.11.0版本开始,在windows上不再支持GPU