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

mac2022-06-30  18

其实,安装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

最新回复(0)