擅长:python、mysql、java
<p>多亏了@Robert Crovella,我解决了我的问题。
原来我只需要使用<code>nvidia/cuda/10.0-devel</code>作为基本映像,而不是<code>nvidia/cuda/10.0-base</code></p>
<p>因此,我的Dockerfile现在是:</p>
<pre><code># Dockerfile
FROM nvidia/cuda:10.0-devel
RUN nvidia-smi
RUN set -xe \
&& apt-get update \
&& apt-get install python3-pip -y \
&& apt-get install git -y
RUN pip3 install upgrade pip
WORKDIR /SingleModelTest
COPY requirements /SingleModelTest/requirements
RUN export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
RUN pip3 install -r requirements/requirements1.txt
RUN pip3 install -r requirements/requirements2.txt
COPY . /SingleModelTest
ENTRYPOINT ["python"]
CMD ["TabNetAPI.py"]
</code></pre>