1、安装cuda
sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-1-local-10.1.105-418.39/7fa2af80.pub
sudo apt-get update
sudo apt-get install -y cuda
sudo gedit ~/.bashrc(写入export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=:/usr/local/cuda-10.1/lib64)
source ~/.bashrc
2\安装cudnn
安装运行时库:
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
安装开发者库:
sudo dpkg -i libcudnn8-dev_7.6.5.32-1+cuda10.1.deb
安装代码示例和cuDNN库文档:
sudo dpkg -i libcudnn8-doc_7.6.5.32-1+cuda10.1.deb
测试
cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
如果cuDNN正确安装,将会打印出如下信息:
Test passed!
3\安装tensorrt
https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-515/tensorrt-install-guide/index.html#installing-tar