- 下载cuda和cudnn,查看显卡型号,下载自己需要的版本。
wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run
sudo sh cuda_11.4.4_470.82.01_linux.run
下载cudnn需要登陆:cudnn-11.4-linux-x64-v8.2.4.15.tgz
注意:安装cuda的时候,不安装driver即显卡驱动。
- cudnn安装
tar -xvf cudnn-11.4-linux-x64-v8.2.4.15.tgz
sudo cp cuda/include/* /usr/local/cuda-11.4/include/
sudo cp cuda/lib64/* /usr/local/cuda-11.4/lib64/
sudo chmod a+r /usr/local/cuda-11.4/include/cudnn.h /usr/local/cuda-11.4/lib64/libcudnn*
- 链接到cuda,如果/usr/local目录以有cuda文件夹先删除。
cd /usr/local
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.4 /usr/local/cuda
- 在~/.bashrc中配置环境,使用zsh就在~/.zshrc中配置环境。
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
5、tensorrt 安装
https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-tar