卸載cuda10、cudnn7.5
查看已安裝cuda
#通過dpkg命令,查看相關cuda安裝的包
dpkg -l | grep -i cuda
查詢出來是這些,可以根據查詢出來的包來用apt-get卸載
sudo apt-get remove cuda-cublas-10-0
sudo apt-get remove cuda-repo-l4t-10-0-local-10.0.166
sudo apt-get remove cuda-toolkit-10-0
sudo apt-get remove cuda-cudart-10-0
sudo apt-get remove cuda-license-10-0
卸載cuda,我這裏一次性卸載
sudo apt-get remove cuda*
查詢cudnn相關已安裝包
dpkg -l | grep -i cudnn
卸載cudnn
sudo apt-get remove libcudnn7*
查詢nvinfer相關已安裝包
dpkg -l | grep -i libnvinfer
卸載nvinfer
sudo apt-get remove libnvinfer*
查詢visionwork相關已安裝包(根據自己需求來選擇卸載)
dpkg -l | grep -i vision
卸載visionwork
sudo apt-get remove libvisio*
查詢opencv相關已安裝包(根據自己需求來選擇卸載)
dpkg -l | grep -i opencv
卸載opencv
sudo apt-get remove libopencv*
下載cuda9、cudnn7.1
可以去tx2的jetpack3.3版本下載相關的安裝包,這裏我已經下載下來了,提供鏈接方便你們下載
cuda9下載鏈接(由於太大 分卷壓縮了):
https://download.csdn.net/download/ourkix/12091647
https://download.csdn.net/download/ourkix/12091663
https://download.csdn.net/download/ourkix/12091665
cudnn7.1下載:
https://download.csdn.net/download/ourkix/12091671
nvinfer下載:
https://download.csdn.net/download/ourkix/12091680
下載完成後文件是這些
cuda-repo-l4t-9-0-local_9.0.252-1_arm64.deb
libcudnn7_7.1.5.14-1+cuda9.0_arm64.deb
libcudnn7-dev_7.1.5.14-1+cuda9.0_arm64.deb
libcudnn7-doc_7.1.5.14-1+cuda9.0_arm64.deb
libnvinfer4_4.1.3-1+cuda9.0_arm64.deb
libnvinfer-dev_4.1.3-1+cuda9.0_arm64.deb
libnvinfer-samples_4.1.3-1+cuda9.0_arm64.deb
libgie-dev_4.1.3-1+cuda9.0_arm64.deb
安裝cuda9、cudnn7.1
根據這些順序來安裝
安裝cuda9(去到下載好解壓的文件夾中,命令行輸入)
sudo dpkg -i cuda-repo-l4t-9-0-local_9.0.252-1_arm64.deb
sudo apt-get install cuda-toolkit-9-0
安裝cudnn
sudo dpkg -i libcudnn7_7.1.5.14-1+cuda9.0_arm64.deb
sudo dpkg -i libcudnn7-dev_7.1.5.14-1+cuda9.0_arm64.deb
sudo dpkg -i libcudnn7-doc_7.1.5.14-1+cuda9.0_arm64.deb
安裝nvinfer
sudo dpkg -i libnvinfer4_4.1.3-1+cuda9.0_arm64.deb
sudo dpkg -i libnvinfer-dev_4.1.3-1+cuda9.0_arm64.deb
sudo dpkg -i libnvinfer-samples_4.1.3-1+cuda9.0_arm64.deb
安裝libgie
sudo dpkg -i libgie-dev_4.1.3-1+cuda9.0_arm64.deb
配置環境變量
修改cuda的環境變量(沒有gedit可以用vim)
sudo gedit ~/.bashrc
打開後,拉倒最下面,將原來的cuda10的環境變量都刪除掉
寫入
export CUBA_HOME=/usr/local/cuda-9.0
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-9.0/bin:/usr/local/bin/cmake:$PATH
保存退出,更新環境變量
source ~/.bashrc
測試
nvcc -V
打印出cuda9.0成功。
接下來可以編譯opencv caffe-ssd等,過程可以看我相關博客。