nvidia-docker运行容器后找不到GPU
执行以下步骤:
1、保证主机已成功安装驱动,执行nvidia-smi
查看。
2、创建容器时使用-e参数添加下面两个环境变量:
NVIDIA_DRIVER_CAPABILITIES=compute,utility
NVIDIA_VISIBLE_DEVICES=all
完整命令为:
docker run -it -e NVIDIA_DRIVER_CAPABILITIES=compute,utility -e NVIDIA_VISIBLE_DEVICES=all $image /bin/bash