由于在最新的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