如何在docker镜像里安装pycuda和numba?

其实，安装numba还好，直接pip install numba就可以。

但pycuda就不那么友好了。

默认安装时，可能会报如下错误：

src/:: fatal error: curand.h: No such file or directory

于是，自己去找了cuda-curand-dev-10-0_10.0.130-1_amd64.deb文件，自己安装了才搞定了。

在以下URL可以找到各个linux发布版的cuda包下载地址：

https://developer.download.nvidia.cn/compute/cuda/repos/

最后，弄一个dockerfile(#注释的忽略，反映了最初的想法)。

FROM harbor.xxx.com.cn/3rd_part/sis_tensorflow:20190812
 
MAINTAINER xxx
 
COPY cuda-curand-dev-10-0_10.0.130-1_amd64.deb /tmp/
 
RUN export http_proxy=http://xxx.local:8080 \
    && export https_proxy=http://xxx.local:8080 \
    && export ftp_proxy=xxx.local:8080 \
    #&& pip install --upgrade pip \
    && dpkg -i /tmp/cuda-curand-dev-10-0_10.0.130-1_amd64.deb \
    #&& apt-get install cuda-cusolver-dev-10-0 \
    #&& apt-get install cuda-curand-dev-10-0 \
    && pip install pycuda \
    && pip install numba \
    && echo "finished!!!"