wsl配置tensorflow环境

问题描述

查阅官网的 tensorflow 版本对照发现Windows 本机的 GPU 支持仅适用于 2.10 或更早版本，从 TF 2.11 开始，Windows 不支持 CUDA 构建。要在 Windows 上使用 TensorFlow GPU，您需要在 WSL2 中构建/安装 TensorFlow，或使用 TensorFlow-DirectML-Plugin 来构建/安装 tensorflow-cpu。

WSL 开发环境配置

WSL 安装

网上教程很多，大致流程差不多：

1. 打开控制面板->程序和功能->启用或关闭 windows 功能，把 适用于 Linux 的 Windows 子系统 和 Hyper-V 勾上
2. cmd 运行 bcdedit /set hypervisorlaunchtype auto
3. Micorsoft Store 安装 Ubuntu

Vscode 配置

下载插件 Remote Development，左上角选择要连接的 WSL

1. 创建工程文件夹 /home/username/project/
2. View->terminal 打开终端，运行以下两条命名：
   sudo apt-get update
sudo apt-get upgrade
3. 安装需要的 python 版本和 pip 工具
4. 创建并启用虚拟环境：
   pyton -m venv venv
source ./venv/bin/acitvate

安装 tensorflow

新版 tf 不用像以前一样找那么多插件的版本对应关系自己安装了。
打开 cmd 用 nvidia-smi 查看驱动版本

+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 536.25                 Driver Version: 536.25       CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+

结果显示我最高能用 cuda12.2。看官网介绍，cuda 12.2 对应 tensorflow 2.15.0。
直接运行 pip install tensorflow[and-gpu]==2.15.0 会报 Could not find TensorRT，手动安装 tensorrt 依然不行。在网上找到解决方案，于是执行：
pip install tensorflow[and-gpu]==2.15.0.post1
执行 python 代码：

import os
import tensorrt
import tensorflow as tf

print(tf.config.list_physical_devices('GPU'))