docker-compose 里 gpu
时间: 2023-07-17 13:54:20 浏览: 88
好的,关于在Docker Compose中使用GPU,您需要进行以下操作:
首先,确保您的主机安装了NVIDIA Docker运行时。您可以通过以下命令来安装:
```
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker
```
接下来,在您的 `docker-compose.yml` 文件中,您需要添加以下内容:
```
version: '3'
services:
app:
build: .
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=all
ports:
- "8080:8080"
```
这个 `docker-compose.yml` 文件中,我们添加了 `runtime: nvidia` 来使用NVIDIA Docker运行时。并且添加了 `NVIDIA_VISIBLE_DEVICES=all` 来指定所有GPU设备可用。
最后,您可以使用以下命令来启动Docker容器:
```
docker-compose up
```
这个命令会根据 `docker-compose.yml` 文件来启动 `app` 服务,并且在主机的 `8080` 端口上暴露您的应用。
希望这些信息可以对您有所帮助,如果您还有其他问题,请随时问我。
阅读全文