【yolov5】yolov5的cuda101 Dockerfile分享

桃扇骨 2023-01-21 02:08 37阅读 0赞

转载请写出处。

1.环境

  1. ubuntu16.04
  2. cuda10.1
  3. cudnn7
  4. opencv3.4
  5. python3.6
  6. torch==1.7.1
  7. torchvision==0.8.2

2.Dockerfile

这里分享一个我写的Dockerfile:

  1. FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu16.04
  2. LABEL maintainer Gu Mengting <1065504814@qq.com>
  3. RUN sed -i s@/archive.ubuntu.com/@/mirrors.aliyun.com/@g /etc/apt/sources.list && \
  4. apt-get clean && \
  5. apt-get -y update
  6. RUN sed -i s@/archive.ubuntu.com/@/mirrors.aliyun.com/@g /etc/apt/sources.list && \
  7. apt-get clean && \
  8. apt-get -y update && apt-get remove x264 libx264-dev && \
  9. apt-get -y install libprotobuf-dev && \
  10. apt-get -y install libleveldb-dev && \
  11. apt-get -y install libsnappy-dev && \
  12. apt-get -y install libopenblas-dev && \
  13. apt-get -y install libopencv-dev && \
  14. apt-get -y install libhdf5-serial-dev && \
  15. apt-get -y install protobuf-compiler && \
  16. apt-get -y install --no-install-recommends libboost-all-dev && \
  17. apt-get -y install libatlas-base-dev && \
  18. apt-get -y install libgflags-dev && \
  19. apt-get -y install libgoogle-glog-dev && \
  20. apt-get -y install liblmdb-dev \
  21. vim \
  22. git \
  23. openssl \
  24. libssl-dev \
  25. wget \
  26. python3-numpy && \
  27. rm -rf /var/lib/apt/lists/*
  28. RUN apt-get -y update && apt-get -y install software-properties-common && \
  29. sh -c '/bin/echo -e "\n" | add-apt-repository ppa:deadsnakes/ppa' && \
  30. apt-get update && apt-get -y install python3.6 \
  31. python3.6-dev \
  32. python3-pip && \
  33. rm -rf /var/lib/apt/lists/*
  34. RUN update-alternatives --install /usr/bin/python python /usr/bin/python2.7 50 && \
  35. update-alternatives --install /usr/bin/python python /usr/bin/python3.5 100 && \
  36. update-alternatives --install /usr/bin/python python /usr/bin/python3.6 150 && \
  37. sh -c '/bin/echo -e "3" | update-alternatives --config python'
  38. RUN python -m pip --default-time=1000 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip && \
  39. python -m pip --default-time=500 install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy \
  40. opencv-python==4.1.0.25 \
  41. matplotlib \
  42. scikit-image && \
  43. rm -rf /var/lib/apt/lists/*
  44. RUN wget https://dl.bintray.com/boostorg/release/1.68.0/source/boost_1_68_0.tar.gz && \
  45. tar -xzvf boost_1_68_0.tar.gz && \
  46. cd boost_1_68_0 && \
  47. sh ./bootstrap.sh --with-python=/usr/bin/python3 --with-python-version=3.6 --with-libraries=all --with-toolset=gcc && \
  48. ./b2 include="/usr/include/python3.6m/" && \
  49. ./b2 install && \
  50. cd /usr/local/lib && \
  51. ln -s libboost_python-py36.so libboost_python3.so && \
  52. ln -s libboost_python-py36.a libboost_python3.a && \
  53. cp /usr/local/lib/libboost_python36.a /usr/lib/x86_64-linux-gnu/libboost_python_python36.a && \
  54. cp /usr/local/lib/libboost_python36.so.1.68.0 /usr/lib/x86_64-linux-gnu/libboost_python3.so && \
  55. cp /usr/local/lib/libboost_python36.so.1.68.0 /usr/lib/x86_64-linux-gnu/ && \
  56. cp /usr/local/lib/libboost_system.so.1.68.0 /usr/lib/x86_64-linux-gnu/ && \
  57. cp /usr/local/lib/libboost_filesystem.so.1.68.0 /usr/lib/x86_64-linux-gnu/ && \
  58. cp /usr/local/lib/libboost_thread.so.1.68.0 /usr/lib/x86_64-linux-gnu/ && \
  59. cd .. && \
  60. rm -rf boost_1_68_0 && \
  61. rm -rf /var/lib/apt/lists/*
  62. ENV OPENCV_ROOT=/opt/opencv
  63. WORKDIR $OPENCV_ROOT
  64. RUN apt-get -y update && apt-get -y install cmake \
  65. unzip && \
  66. wget https://codeload.github.com/opencv/opencv/zip/3.4 -O opencv-3.4.zip && unzip opencv-3.4.zip && cd opencv-3.4 && \
  67. mkdir build && cd build && \
  68. cmake -D CMAKE_BUILD_TYPE=RELEASE \
  69. -D PYTHON_DEFAULT_EXECUTABLE=/usr/bin/python \
  70. -D PYTHON_INCLUDE_DIR=/usr/include/python3.6 \
  71. -D PYTHON_INCLUDE_DIR2=/usr/include/x86_64-linux-gnu/python3.6m \
  72. -D PYTHON_LIBRARY=/usr/lib/x86_64-linux-gnu/libpython3.6m.so \
  73. -D PYTHON3_NUMPY_INCLUDE_DIRS=/usr/lib/python3/dist-packages/numpy/core/include/ \
  74. -D PYTHON3_EXECUTABLE=/usr/lib/python \
  75. -D BUILD_opencv_python2=OFF \
  76. -D CMAKE_INSTALL_PREFIX=/usr/local \
  77. -D INSTALL_C_EXAMPLES=ON \
  78. -D INSTALL_PYTHON_EXAMPLES=ON \
  79. -D WITH_TBB=ON \
  80. -D WITH_V4L=ON \
  81. -D WITH_OPENGL=ON \
  82. -D BUILD_EXAMPLES=ON .. && make -j"$(nproc)" && make install && \
  83. rm -rf /var/lib/apt/lists/*
  84. RUN echo "/usr/local/lib" >> /etc/ld.so.conf.d/opencv.conf && ldconfig
  85. ENV WORKSPACE_ROOT=/root/
  86. WORKDIR $WORKSPACE_ROOT
  87. RUN python -m pip --default-time=500 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade numpy &&\
  88. python -m pip --default-time=500 install -i https://pypi.tuna.tsinghua.edu.cn/simple Cython \
  89. matplotlib>=3.2.2 \
  90. numpy>=1.18.5 \
  91. opencv-python==4.1.0.25 \
  92. Pillow \
  93. PyYAML>=5.3 \
  94. scipy>=1.4.1 \
  95. tensorboard>=2.2 \
  96. torch>=1.7.0 \
  97. torchvision>=0.8.1 \
  98. tqdm>=4.41.0 \
  99. seaborn>=0.11.0&&\
  100. rm -rf /var/lib/apt/lists/*

发表评论

表情:
评论列表 (有 0 条评论,37人围观)

还没有评论,来说两句吧...

相关阅读