开发环境
- windows
使用工具
- docker-destop
- wsl2
安装NV驱动
Install the NVIDIA GPU driver for your Linux distribution
官方文档
Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkit 1.16.2 documentation
配置Docker
Setting –> Docker Engine
修改json数据
{
"runtimes": {
"nvidia": {
"path": "nvidia-container-runtime",
"runtimeArgs": []
}
}
}
重启服务
测试GPU支持
docker run --gpus all nvidia/cuda:11.0-base nvidia-smi