其实,安装numba还好,直接pip install numba就可以。
但pycuda就不那么友好了。
默认安装时,可能会报如下错误:
src/cpp/curand.hpp:6:12: 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!!!"
转载于:https://www.cnblogs.com/aguncn/p/11418377.html