Tag: 咖啡

用Dockerfile安装咖啡时出错

我试图通过这个Dockerfile安装caffe – > https://github.com/floydhub/dl-docker/blob/master/Dockerfile.gpu 我使用以下命令在计算机上构builduser@maquina:/homeLocal/hugo/docker1/dl-docker⟫ docker build -t floydhub/dl-docker:gpu -f Dockerfile.gpu . : user@maquina:/homeLocal/hugo/docker1/dl-docker⟫ docker build -t floydhub/dl-docker:gpu -f Dockerfile.gpu . 但是,在施工期间,会发生以下错误: The command '/bin/sh -c git clone -b ${CAFFE_VERSION} –depth 1 https://github.com/BVLC/caffe.git /root/caffe && cd /root/caffe && cat python/requirements.txt | xargs -n1 pip install && mkdir build && cd build && cmake -DUSE_CUDNN=1 -DBLAS=Open […]

在成功安装GPU Caffe后,将Caffe作为CPU运行

我有一个用cudnn支持编译的Caffe的Docker镜像。 CUDA和所有其他依赖项在映像中正确安装,使用nvidia-docker从主机提供驱动程序时,它可以正常工作。 我想在不使用nvidia-docker 情况下运行相同的映像,只需将Caffe设置为CPU模式即可。 但是,当我这样做,我仍然看到错误,正确的Nvidia驱动程序无法find。 就像用cudnn支持来构buildCaffe导致Caffe 需要 GPU驱动程序一样。 这对于我的用例来说是有问题的:提供一个单独的Docker镜像,它可以交替使用,既可以是基于CPU的镜像,也可以是基于GPU的镜像。 我如何在启用cudnn / GPU支持的情况下安装Caffe,但是仍然运行它,而不依赖于任何CUDA / GPU依赖关系? 注意:这不是在Caffe构build时使用CPU_ONLY标志禁止GPU支持的问题。 相反,它需要将Caffe构build成能够使用GPU,然后以一种不需要任何驱动程序,CUDA库等的方式,在仅CPU 运行模式下运行它 。 我在运行时看到的错误types如下: I0523 21:09:42.175459 14 layer_factory.hpp:77] Creating layer conv1 I0523 21:09:42.175561 14 net.cpp:84] Creating Layer conv1 I0523 21:09:42.175606 14 net.cpp:406] conv1 <- data I0523 21:09:42.175660 14 net.cpp:380] conv1 -> conv1 F0523 21:09:42.177079 14 cudnn_conv_layer.cpp:52] Check failed: error == […]

为caffe构buildGPU docker镜像时出错:不支持的gpu架构“compute_60”

尝试构build基于gpu的caffe docker镜像失败。 该命令运行: docker build -t caffe:gpu standalone/gpu 它运行了大约二十分钟,包括通过cython编译许多numpy / scikit脚本到c代码。 然而这个版本最终在nvidia nvcc步骤中失败了: [ 1%] [ 1%] Building NVCC (Device) object src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_cudnn_pooling_layer.cu.o Building NVCC (Device) object src/caffe/CMakeFiles/cuda_compile.dir/util/./cuda_compile_generated_im2col.cu.o nvcc fatal : Unsupported gpu architecture 'compute_60' CMake Error at cuda_compile_generated_cudnn_pooling_layer.cu.o.cmake:206 (message): Error generating /opt/caffe/build/src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_cudnn_pooling_layer.cu.o make[2]: *** [src/caffe/CMakeFiles/cuda_compile.dir/layers/./cuda_compile_generated_cudnn_pooling_layer.cu.o] Error 1 make[2]: *** Waiting for unfinished jobs…. nvcc fatal […]