參考https://blog.csdn.net/xjgao75/article/details/105869699/#UbuntuNVDIA_52
一. 安裝nvidia 驅(qū)動(dòng)
下載離線安裝包 NVIDIA-Linux-x86_64-440.82.run
1.禁用nouveau:命令如下:
sudo chmod 666 /etc/modprobe.d/blacklist.conf
sudo vim /etc/modprobe.d/blacklist.conf
2.在最后面添加如下代碼:
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
3.然后保存(esc鍵谣辞,然后shift鍵+:鍵淤袜,然后輸入wq捉超,回車)
為了保證完全禁用掉笋粟,還要在內(nèi)核中直接禁用Nouveau:
命令如下:
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
完成后再把權(quán)限改回出來招驴,命令如下:
sudo chmod 644 /etc/modprobe.d/blacklist.conf
這之后侵状,更新配置試生效试和,命令如下:
sudo update-initramfs -u
4.重啟:sudo reboot
重啟完成后店读,我們需要關(guān)閉圖形界面,如果沒有關(guān)閉圖形界面峭弟,開始安裝附鸽,測試時(shí)會(huì)報(bào)錯(cuò)。
ctrl+alt+f1 切換到終端界面
關(guān)閉圖形界面命令如下:
service lightdm stop
5. 開始安裝
sudo ./NVIDIA-Linux-x86_64-440.82.run -no-x-check -no-nouveau-check -no-opengl-files
安裝完成
(也可以通過附加驅(qū)動(dòng)安裝 簡單)
二. 安裝cuda
下載離線安裝包 cuda_10.2.89_440.33.01_linux.run
sudo ./cuda_10.2.89_440.33.01_linux.run
開始安裝
accept
取消選中驅(qū)動(dòng)安裝
install
完成
查看是否安裝成功
nvidia-smi
下一步瞒瘸,配置變變量環(huán)境:
執(zhí)行命令:
vim ~/.bashrc
添加如下代碼:
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
然后保存
輸入命令:
source ~/.bashrc
重啟:
sudo reboot
開始測試cuda是否安裝成功:
命令如下:
cd /usr/local/cuda-10.2/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
三. 安裝cudnn
下載離線安裝包
https://developer.nvidia.com/cudnn
libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb
四. 安裝conda
下載 Anaconda3-4.2.0-Linux-x86_64.sh
https://repo.continuum.io/archive/Anaconda3-4.2.0-Linux-x86_64.sh
或https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
安裝bash Anaconda3-4.2.0-Linux-x86_64.sh
一路安裝... 并加入環(huán)境變量
五. 離線安裝pytorch+torchvision
下載離線安裝包 pytorch-1.5.0-py3.5_cuda10.2.89_cudnn7.6.5_0.tar.bz2
下載離線安裝包 torchvision-0.6.0-py35_cu102.tar.bz2
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
1. 安裝torch
在同一文件夾下運(yùn)行:
conda install pytorch-1.5.0-py3.5_cuda10.2.89_cudnn7.6.5_0.tar.bz2
2. 再安裝torchvision :
conda install torchvision
最終失敗
通過 安裝也失敗pip install torchvision
最后還是通過離線安裝
conda install torchvision-0.6.0-py35_cu102.tar.bz2
安裝Pytorch之后坷备,輸入import torch,出現(xiàn)ImportError: libmkl_gnu_thread.so: cannot open shared object file: No such file or directory
首先,嘗試把相關(guān)的so文件添加到路徑里,ldconfig 更新一下,不能解決問題.
1.使用如下命令debug一下
conda install --debug mkl
2.使用如下命令重新裝一下mkl包
conda install mkl
原文鏈接:https://blog.csdn.net/qq_22704577/java/article/details/53928059
測試
進(jìn)入python交互式環(huán)境
python
import torch
print(torch.__version__)
六 opencv源碼安裝
安裝依賴
必須要安裝的依賴情臭,通過命令安裝:
sudo apt-get install build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
如果需要其他的功能省撑,還需安裝一些,命令:
sudo apt-get install python-dev python-numpy python3-dev python3-numpy
sudo apt-get install libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev
安裝OpenCV
下載OpenCV的安裝包俯在,下載Sources包竟秫,opencv-4.2.0.tar.gz 解壓:
tar -zxvf opencv-4.2.0.tar.gz
在解壓的文件夾里新建一個(gè)文件夾用來編譯OpenCV:
cd opencv-4.0.1
mkdir release
cd release
然后在新建的這個(gè)文件里運(yùn)行cmake:
cmake -DCMAKE_BUILD_TYPE=Release \
-DOPENCV_GENERATE_PKGCONFIG=ON \
-DCMAKE_INSTALL_PREFIX=/usr/local ..
OpenCV4默認(rèn)不生成.pc文件,OPENCV_GENERATE_PKGCONFIG=ON才會(huì)生成跷乐。
之后就是編譯和安裝了肥败,我的電腦是四核八線程,可以運(yùn)行make -j8,但我不想編譯時(shí)影響我工作馒稍,所以用-j7皿哨,如果是四核可用make -j4。
make -j7
sudo make install
如果你安裝了python-dev,這時(shí)Python應(yīng)該可以使用OpenCV了纽谒,在終端輸入python证膨,可以查看OpenCV版本:
$ python
>>> import cv2
>>> cv2.__version__
'4.0.1'
>>>
通過,安裝Python-OpenCV解決
pip install opencv-python
之后就是配置c++了,先在/etc/ld.so.conf.d/文件夾下新建一個(gè)opencv4.conf鼓黔,里面寫入/usr/local/lib央勒。
cd /etc/ld.so.conf.d/
sudo touch opencv4.conf
sudo sh -c 'echo "/usr/local/lib" > opencv4.conf'
更新pkg-config:
sudo ldconfig
解決報(bào)錯(cuò)1:darknet-yolo:fatal error: opencv2/opencv.hpp: No such file or directory
sudo apt install libopencv-dev
解決報(bào)錯(cuò)2:/bin/sh: 1: nvcc: not found make: *** [obj/convolutional_kernels.o] Error 127
修改makefile
NVCC = /usr/local/cuda-10.2/bin/nvcc
解決報(bào)錯(cuò)3:cuda error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0' failed
那么,修改cfg文件夾下的yolov3.cfg文件请祖,原始的yolov3.cfg文件開頭為:
[net]
# Testing
#batch=1
#subdivisions=1
# Training
batch=64
subdivisions=16
修改為:
[net]
# Testing
batch=1
subdivisions=1
# Training
#batch=64
#subdivisions=16