带有NVIDIA运行时的Docker构建,例如“ --gpus”标志

时间:2019-12-25 03:58:33

标签: docker gpu

我最近为机器学习项目构建了docker容器,该容器需要根据NVIDIA运行时使用一些库进行构建。
我想用docker build准备完全设置的容器,但是失败了,因为在docker build期间不提供NVIDIA运行时,该运行时不提供--gpus标志。
docker build时有什么方法可以使用NVIDIA运行时?

0 个答案:

没有答案