使用以下命令运行最新的docker: docker run -it -p 8888:8888 tensorflow/tensorflow:latest-gpu-jupyter jupyter notebook/cuda/cuda_diagnostics.cc:163] no NVIDIA GPU device is present: /dev/nvidia0 does not exist 我使用的是CU
在启动使用GPU的Docker容器的情况下,如何创建气流任务。当从终端运行时,我只会使用--gpus all标志。DockerOperator, because it does not support device\_requests parameter, which is used underneath when calling docker
终端-它的工作:docker run -it --rm --gpus=all --entrypoint="/bin/bash" 3b6d609a5189# image has an entrypoint, so I overwrite it
在容器中,我可以运行nvidia-smi来查看GPU是否被找到,并确认Tensorflow找到了它注意:使用doc