当前位置:   article > 正文

WSL2配置深度学习环境,cuda+cudnn+opencv_wsl cuda

wsl cuda

一、cuda

1、安装cuda

安装cuda比较简单,进入官网,选择自己需要的版本,按照命令运行即可

CUDA Toolkit Archive | NVIDIA Developer

需要注意的是,选择WSL-Ubuntu

img

2、配置环境变量

sudo vim ~/.bashrc

移动到最后一行,shift+insert可以复制到vim中。

  1. export CUDA_HOME=/usr/local/cuda-12.2
  2. export PATH=/usr/local/cuda-12.2/bin:$PATH
  3. export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PAT

最后按ESC退出编辑,输入“:wq”,或者 “ZZ” 保存。

然后输入以下命令使环境变量生效:

source ~/.bashrc

3、测试

nvcc -V

二、cudnn

1、进入官网进行下载,选择匹配自己cuda版本

cuDNN Archive | NVIDIA Developer

2、解压文件,并将其放入cuda对应的文件中

  1. tar -xvf cudnn-linux-x86_64-8.9.3.28_cuda12-archive.tar.xz
  2. sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda-12.2/include
  3. sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda-12.2/lib64
  4. sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda-12.2/lib64/libcudnn*

3、测试

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

三、编译CUDA12.2版的opencv

1、安装依赖项

  1. sudo apt-get install cmake git
  2. sudo apt-get install build-essential libgtk2.0-dev libavcodec-dev libavformat-dev libjpeg-dev libswscale-dev libtiff5-dev libgtk2.0-dev pkg-config

2、下载OpenCV、OpenCV_contrib

3、编译安装

  1. mkdir build
  2. cd build
  3. cmake -D CMAKE_BUILD_TYPE=RELEASE \
  4. -D CMAKE_INSTALL_PREFIX=/usr/local \
  5. -D INSTALL_C_EXAMPLES=ON \
  6. -D OPENCV_ENABLE_NONFREE=ON \
  7. -D WITH_CUDA=ON \
  8. -D WITH_CUDNN=ON \
  9. -D WITH_TBB=ON \
  10. -D OPENCV_DNN_CUDA=ON \
  11. -D ENABLE_FAST_MATH=1 \
  12. -D CUDA_FAST_MATH=1 \
  13. -D WITH_CUBLAS=1 \
  14. -D OPENCV_GENERATE_PKGCONFIG=ON \
  15. -D OPENCV_EXTRA_MODULES_PATH=/mnt/f/project_linux/opencv_contrib-4.8.1/modules \
  16. -D WITH_WEBP=OFF \
  17. -D WITH_OPENCL=OFF \
  18. -D ETHASHLCL=OFF \
  19. -D ENABLE_CXX11=ON \
  20. -D BUILD_EXAMPLES=OFF \
  21. -D OPENCV_ENABLE_NONFREE=ON \
  22. -D WITH_OPENGL=ON \
  23. -D WITH_GSTREAMER=ON \
  24. -D WITH_V4L=ON \
  25. -D WITH_QT=OFF \
  26. -D BUILD_opencv_python3=ON \
  27. -D BUILD_opencv_python2=OFF \
  28. -D HAVE_opencv_python3=ON   ..
  29. make -j8
  30. sudo make install

4、一些错误

sudo apt install libopenjp2-7-dev

cuda版本太高,下载opencv4.8.1版本

解决办法:Ubuntu20 安装 带cuda的opencv遇到的问题_cuda4dnn/primitives/normalize_bbox.hpp: in instant-CSDN博客

You want to change line 114 in opencv/modules/dnn/src/cuda4dnn/primitives/normalize_bbox.hpp: from: if (weight != 1.0) to: if (weight != static_cast<T>(1.0))

As well as line 124 in opencv/modules/dnn/src/cuda4dnn/primitives/region.hpp (due to a similar error): from: if (nms_iou_threshold > 0) { to: if (nms_iou_threshold > static_cast<T>(0)) {

参考:

Linux下编译Opencv和contrib_linux opencv_contrib-4.5.2编译-CSDN博客

Linux系统编译CUDA版的OpenCV4.2_linux c++使用opencv_contrib dnn-CSDN博客

Install Caffe on Ubuntu 20.04 with OpenCV 4.4 - Q-engineering (qengineering.eu)

ubuntu 编译安装支持CUDA的OpenCV+其他opencv细节(需要特定版本gcc)_ubuntu opencv cuda-CSDN博客

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/927213
推荐阅读
相关标签
  

闽ICP备14008679号