Question

尝试安装具有GPU支持的dlib。使用tf图片

FROM tensorflow/tensorflow:latest-gpu-py3

安装cudnn

COPY cudnn-9.0-linux-x64-v7.1.tgz ./
RUN tar -zxvf cudnn-9.0-linux-x64-v7.1.tgz
RUN cp -P cuda/include/cudnn.h /usr/local/cuda/include
RUN cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
RUN chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

然后安装dlib

RUN git clone https://github.com/davisking/dlib
RUN cd dlib && python3 setup.py install

我收到此错误

[1％]构建NVCC（设备）对象   dlib_build / CMakeFiles / dlib.dir / cuda / dlib_generation_cuda_dlib.cu.o [   1％]建立NVCC（装置）物件   dlib_build / CMakeFiles / dlib.dir / cuda / dlib_generation_cusolver_dlibapi.cu.o   /notebooks/dlib/dlib/cuda/cusolver_dlibapi.cu:9:23：致命错误：   cublas_v2.h：此类文件或目录的编译没有终止。 CMake的   dlib_generated_cusolver_dlibapi.cu.o.cmake出现错误：207（消息）：
  生成错误
  /notebooks/dlib/build/temp.linux-x86_64-3.5/dlib_build/CMakeFiles/dlib.dir/cuda/。/dlib_genic_cusolver_dlibapi.cu.o

如果我不安装cudnn，则dlib会成功构建，但会通知它不使用CUDA。

尝试过使用google，但是人们写来设置USE_CPU_ONLY = 1（他们有caffe错误），这不是我所需要的。

Answer 1

遇到了同样的问题。安装cublas-dev（sudo apt install cuda-cublas-dev-9-0）为我解决了此问题。

找不到安装dlib错误cublas_v2

1 个答案: