
本文共 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 ������������������������������������������
发表评论
最新留言
关于作者
