我正在尝试从Pycharm IDE内部使用tensorflow提供的启用gpu的docker映像运行一个简单的tensorflow示例。一切正常,除了运行时,tensorflow不会从容器中检测到GPU,而是默认返回到CPU:
tensorflow/stream_executor/cuda/cuda_driver.cc:318] failed call to cuInit:
Run Code Online (Sandbox Code Playgroud)
运行:Ubuntu 18.04.3 Docker 19.03.3最新版本的NVIDIA Docker支持如下:https : //github.com/NVIDIA/nvidia-docker
我已经使用远程解释器功能设置了Pycharm项目以运行image:tensorflow:latest-gpu
如果我使用以下命令从命令行运行容器:
docker run --gpus all --rm tensorflow/tensorflow:latest-gpu nvidia-smi
Run Code Online (Sandbox Code Playgroud)
我得到这个:
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.26 Driver Version: 430.26 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 108... Off | 00000000:41:00.0 On | N/A |
| 28% 26C P8 9W / 250W | …Run Code Online (Sandbox Code Playgroud)