[英]CUDA cudaErrorInsufficientDriver error when running an CuFFT in docker
我在 docker 中做一些 CUFFT 例程並遇到了一些問題。 我使用以下 Dockerfile。
FROM nvidia/cuda:9.1-runtime-ubuntu16.04
ENV NVIDIA_VISIBLE_DEVICES all
ENV LD_LIBRARY_PATH /usr/local/cuda-9.1/lib64/
FROM python:3.7
COPY --from=0 /usr/local/cuda-9.1 /usr/local/cuda-9.1
ENV VIRTUAL_ENV=/opt/venv
ENV PATH="/opt/venv:$PATH"
RUN pip install numpy
RUN apt update && \
apt-get -y install gcc && \
apt-get -y install apt-utils && \
apt-get -y install g++ && \
apt-get -y install pciutils && \
apt-get -y install libc6
ADD helmsolver /helmsolver
CMD ls /usr && ls /usr/local
CMD dpkg -l | grep -i cuda
CMD cd helmsolver && bash tests.sh
要構建和運行 docker 我使用這樣的命令。
docker build -t helm .
docker run --gpus all helm
我可以在主機上運行我的代碼,但是在 docker 中運行后,錯誤 35 ( cudaErrorInsufficientDriver
) 出現在這種類型的代碼cudaMalloc((void**)&d_array, memsize)
中。 我的代碼有什么問題,還是只是缺少一些.so
文件? 這是我的 CUDA、docker、nvidia-smi 版本
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85
Docker version 19.03.4
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.67 Driver Version: 418.67 CUDA Version: N/A |
|-------------------------------+----------------------+----------------------+
| 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 GT 640 On | 00000000:01:00.0 N/A | N/A |
| 40% 36C P8 N/A / N/A | 48MiB / 4035MiB | N/A Default |
+-------------------------------+----------------------+----------------------+
| 1 GeForce GTX 760 On | 00000000:02:00.0 N/A | N/A |
| 17% 36C P8 N/A / N/A | 1MiB / 4037MiB | N/A Default |
+-------------------------------+----------------------+----------------------+
添加NVIDIA_DRIVER_CAPABILITIES compute, utility
作為 ENV 解決了該問題。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.