Windows下Tensorflow docker python开发环境搭建

前置条件

windows10 更新到较新的版本，硬件支持Hyper-V。

参考：https://learn.microsoft.com/zh-cn/windows/wsl/install

启用WSL

在Powershell中输入如下指令：

dism.exe /online /enable-feature /featurename:Microsoft-Windows-Subsystem-Linux /all /norestart

启用虚拟机平台

Enable-WindowsOptionalFeature -Online -FeatureName VirtualMachinePlatform -NoRestart

设置wsl2 为默认值

wsl --set-default-version 2

下载docker

https://docs.docker.com/desktop/install/windows-install/

下载一个image

启动docker

具体参考：https://www.tensorflow.org/install/docker?hl=zh-cn

docker run --gpus all -it --rm tensorflow/tensorflow:latest-gpu

然后可以在容器中运行python代码，所有环境都已经配置好，仅需要主机安装好显卡驱动。

安装Windows Terminal（可选）

Win11应该是自带的，win10需要

在vscode中安装远程开发扩展

使用vscode连接docker

在远程主机中安装python扩展

进行代码调试