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