深度学习环境安装
发布日期:2021-05-10 23:43:29 浏览次数:18 分类:精选文章

本文共 5058 字,大约阅读时间需要 16 分钟。

Ubuntu 18.04 ��������� Python 3.6���GPU ��� OpenCV 4.1 ������������

������������������������������������������ Ubuntu 18.04 ��������������������������� Python 3.6���NVIDIA GPU ��� OpenCV 4.1 ���������������������������������������������������������������������������������������������������

1. ���������������������

���������������������������������������������������������������������

sudo apt-get install build-essential cmake unzip pkg-configsudo apt-get install libjpeg-dev libpng-dev libtiff-devsudo apt-get install libavcodec-dev libavformat-dev libswscale-dev libv4l-devsudo apt-get install libxvidcore-dev libx264-devsudo apt-get install libgtk-3-devsudo apt-get install libatlas-base-devsudo apt-get install gfortransudo apt-get install python3-dev

��������������������� OpenCV ��� CUDA desarrollo���������������������

2. ������ Python ������

��������������������������� Python 3.6 ������������������

sudo apt-get install python3sudo apt-get install python3-pipsudo apt-get install python3-tksudo apt-get install python3-numpysudo apt-get install python3-scipysudo apt-get install python3-matplotlibsudo apt-get install python3-pandas

������������������������������ python3 --version ������ Python ��������������� 3.6 ������������������

3. ������ OpenCV 4.1

OpenCV ������������������������������������������������������������������������������������ OpenCV 4.1���

# ������ OpenCV ������git clone https://github.com/opencv/opencv.git# ������ OpenCV ������cd opencv4.1.0# ������ build ������������������mkdir buildcd build# ������ CMake ������������cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local ..# ���������������sudo make -j8sudo make install

������������������������������ python3 ��� opencv-python ��������� OpenCV ���������������������

4. ������ NVIDIA ������������

��� Ubuntu ������������NVIDIA ������������������������������������������������������������������

���������������������

������������������������������������������ Nouveau ���������

sudo vim /etc/modprobe.d/blacklist.conf

���������������������������������

blacklist nouveauoptions nouveau modeset=0

���������������������������������������������������������������

lspci | grep -i nvidia

������������������ NVIDIA ������������������������������

������ NVIDIA ������

��������������������������������������������������������������� NVIDIA 430 ���������

sudo apt-get install nvidia-driver-430

���������������������������������

������������������

������������������������ NVIDIA ���������������������������

nvidia-smi

��������������������� NVIDIA ������������������

5. ������ CUDA 9.0 ������������

CUDA 9.0 ������������������ GPU ������������������������������������������ CUDA���

������ CUDA ������������

��� NVIDIA ������������������ CUDA 9.0 ���������������������������������������������

# ������������������# ��������������������� CUDA ������sudo ./cuda_9.0.176_384.81_linux.run

������ CUDA ���Ok���������

��������������������������������� cuda-9.0 ���������������������

sudo mkdir -p /usr/local/cuda-9.0sudo mount /dev/nvme1 /usr/local/cuda-9.0

������������������

���������������������������������������

  • ������ EULA ������������������
  • ������������ CUDA ������������
  • ������������������ uncommentally ������ CUDA ������������������

������������������������ CUDA ������������������������������ /usr/local/cuda-9.0���

6. ������ CUDNN 7.5

CUDNN ��� CUDA ������������������������������������������������������������

������ CUDNN ���������

��� CUDNN ������������������������������������������������

sudo cp -r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/include/sudo cp /usr/local/cuda-9.0/lib64/libcudnn* /usr/local/cuda-9.0/lib64/sudo chmod -r 777 /usr/local/cuda-9.0/include/cudnn.h sudo chmod -r 777 /usr/local/cuda-9.0/lib64/libcudnn*

������������

������ CUDNN ���������������������

cat /usr/local/cuda-9.0/version.txtcat /usr/local/cuda-9.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

������������ CUDNN ������������������

7. ������ CUDA ������������

��������������������������������� CUDA ������������������

������Shell ������������

��� .bashrc ��������������� CUDA ������������������

# CUDA_HOME ���������������������������export CUDA_HOME=/usr/local/cuda-9.0export PATH=/usr/local/cuda-9.0/bin:"${PATH}"export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:""LD_LIBRARY_PATH

������ CUDA ������

���������������������������

source ~/.bashrc

������������������������������������������������ Python 3.6���GPU ��������� OpenCV 4.1 ���������������

8. ������������������

���������������������������������������������������������������������������������������������������������

TensorFlow GPU ������

������ TensorFlow ��� GPU ���������

pip3 install tensorflow-gpu==1.6.0pip3 install keras==2.1.5

PyTorch

������ PyTorch���

pip3 install torchpip3 install torchvision

���������������

������������������������

pip3 install h5pypip3 install pillowpip3 install visdom

������������������������������������������������������������������������������������������������������������������������������������������������������ Docker ������������������������������������������

上一篇:物体检测模型评价
下一篇:解决Vue打包后放到部署到服务器运行时出现Loading chunk * failed 错误

发表评论

最新留言

很好
[***.229.124.182]2025年04月26日 04时29分14秒