当我使用 tensorflow_gpu 时,Dlib 不使用 GPU
Dlib don't using GPU when I using tensorflow_gpu
我有 2 个版本 的 tensorflow、cpu 和 gpu。我正在使用 conda 来管理我的环境。当我激活 cpu 时,dlib 可以在 GPU 上很好地 运行。但是当我使用 gpu 时,它只是 cpu 上的 运行。我不知道为什么,请帮助我。
有一些我的系统信息,不知道能不能给点信息。
- Ubuntu 18.04LTS
- python 3.6
- CUDA 10.1
- tensorflow-GPU 1.9.0
- 张量流 1.12.0
测试代码
conda activate ML
ipython
>>> import dlib
>>> dlib.DLIB_USE_CUDA
True
上面的代码输出 True 但下面的代码输出 False。
conda activate ML_GPU
ipython
>>> import dlib
>>> dlib.DLIB_USE_CUDA
False
我希望当我使用 GPU 版本的 tensorflow dlib 时,可以像 CPU 那样使用我的 GPU。有人可以帮忙吗?
我已经通过手动安装 dlib 解决了这个问题。
有一些步骤可以做到这一点。
注意:您应该在虚拟环境中执行这些步骤。
conda activate ML_GPU
pip uninstall dlib
git clone https://github.com/davisking/dlib.git
cd dlib
mkdir build
cd build
cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1
cmake --build .
cd ..
python setup.py install
最后一步还是有问题。如果您看到这样的信息,您应该注意——已找到 CUDA,但您的编译器未能编译一个简单的 CUDA 程序,因此 dlib 不会使用 CUDA。
那么你应该改变你的g++编译器版本。你也可以从
找一些资料
。
我的g++版本是:(Ubuntu 6.5.0-2ubuntu1~18.04) 6.5.0 20181026,最新版本似乎与CUDA不兼容。
如果一切正常,你会看到
>>> dlib.DLIB_USE_CUDA
True
我有 2 个版本 的 tensorflow、cpu 和 gpu。我正在使用 conda 来管理我的环境。当我激活 cpu 时,dlib 可以在 GPU 上很好地 运行。但是当我使用 gpu 时,它只是 cpu 上的 运行。我不知道为什么,请帮助我。
有一些我的系统信息,不知道能不能给点信息。
- Ubuntu 18.04LTS
- python 3.6
- CUDA 10.1
- tensorflow-GPU 1.9.0
- 张量流 1.12.0
测试代码
conda activate ML
ipython
>>> import dlib
>>> dlib.DLIB_USE_CUDA
True
上面的代码输出 True 但下面的代码输出 False。
conda activate ML_GPU
ipython
>>> import dlib
>>> dlib.DLIB_USE_CUDA
False
我希望当我使用 GPU 版本的 tensorflow dlib 时,可以像 CPU 那样使用我的 GPU。有人可以帮忙吗?
我已经通过手动安装 dlib 解决了这个问题。 有一些步骤可以做到这一点。
注意:您应该在虚拟环境中执行这些步骤。
conda activate ML_GPU
pip uninstall dlib
git clone https://github.com/davisking/dlib.git
cd dlib
mkdir build
cd build
cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1
cmake --build .
cd ..
python setup.py install
最后一步还是有问题。如果您看到这样的信息,您应该注意——已找到 CUDA,但您的编译器未能编译一个简单的 CUDA 程序,因此 dlib 不会使用 CUDA。
那么你应该改变你的g++编译器版本。你也可以从
找一些资料
我的g++版本是:(Ubuntu 6.5.0-2ubuntu1~18.04) 6.5.0 20181026,最新版本似乎与CUDA不兼容。
如果一切正常,你会看到
>>> dlib.DLIB_USE_CUDA
True