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

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM