win10 wsl2 docker

由于在最新的win10 发行版中,更新的wsl2支持了GPU cuda 计算。而在最新的docker 中也支持了将wsl2 作为后端引擎,

这样一来,直接在win10 下安装docker,并下载启动镜像就能调用GPU。

只能说这一波操作太6, 微软牛逼,docker 牛逼!

具体做法如下,首先升级win10 到最新版,接着安装从docker 官网安装docker for windows

安装完成之后,打开docker 然后选择

即可。

随后即可在命令行中用docker 命令拉去docker镜像

docker pull  nvidia:10.2

然后启动

使用下面的命令
docker run -d -it --name="centos"  --gpus all mytorch:v1 /bin/bash

 即可调用到GPU,

随后安装pytorch

pip install torch