nvidia cuda cudnn tensorrt安装配置

1 篇文章 0 订阅
1 篇文章 0 订阅
  1. 下载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即显卡驱动。

  1. 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*
  1. 链接到cuda,如果/usr/local目录以有cuda文件夹先删除。
cd /usr/local
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.4 /usr/local/cuda
  1. 在~/.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
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值