只能使用支持 cuda 的 nvidia 显卡,其他不行😭
docker run
https://docs.docker.com/config/containers/resource_constraints/#gpu
--gpus all
docker run -it --rm --gpus all nvidia/cuda:12.3.1-base-ubuntu20.04 nvidia-smi
docker compose
https://docs.docker.com/compose/gpu-support/
services:
test:
image: nvidia/cuda...