安装nvidia drivers, cuda toolkit, cudnn, tensorrt

在RTX3070的机器上安装nvidia深度学习大礼包

nvidia driver 安装

ubuntu-drivers devices

driver   : nvidia-driver-510 - third-party non-free recommended
driver   : nvidia-driver-465 - third-party non-free
driver   : nvidia-driver-460 - third-party non-free
driver   : nvidia-driver-470-server - distro non-free
driver   : nvidia-driver-470 - third-party non-free
driver   : nvidia-driver-495 - third-party non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

sudo apt-get install nvidia-drivers-470

cuda toolkit安装
可以选择.run或者.deb的方式安装,我尝试过.deb的方式安装,会自动更新我的driver到最新版本,自动安装最新的cuda版本,所以我这里选择.run的方式安装。

官方安装教程
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
下载地址
https://developer.nvidia.com/cuda-toolkit-archive

wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run

在这里插入图片描述
cudnn 安装
下载地址
https://developer.nvidia.com/cudnn
官方安装教程
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

Navigate to your directory containing the cuDNN tar file.
Unzip the cuDNN package.

$ tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-archive.tar.xz

Copy the following files into the CUDA toolkit directory.

$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

tensorrt 安装
如果cuda toolkit是.deb安装的,那么tensorrt也可以用.deb安装。如果cuda toolkit是.run安装的,那么这里就需要下载tar压缩包进行安装。

下载地址
https://developer.nvidia.com/nvidia-tensorrt-7x-download
官方安装教程
https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值