执行以下步骤:
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
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
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。