linux16.04安装cuda和cudnn, TensorRT

骑猪看日落 2022-05-11 09:06 223阅读 0赞

## 安装cuda ##

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
    sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
    sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
    sudo apt-get update
    sudo apt-get install cuda-9-0 -y
    sudo nvidia-xconfig

配置环境路径：  
vim ~/.bashrc

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/lib/:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
    export CUDA_HOME="/usr/local/cuda"
    export PATH="$CUDA_HOME/bin:$HOME/bin:$HOME/.local/bin:/usr/bin:/usr/local/bin:$PATH"
    export LD_LIBRARY_PATH="/usr/local/openmpi/lib/:$LD_LIBRARY_PATH"

执行 source ~/.bashrc

重启机器之后，nvidia-smi生效

## 安装cudnn ##

在https://developer.nvidia.com/rdp/cudnn-archive 下载适合的版本的cudnn:  
在16.04 x86版本我下载的是：cudnn-9.0-linux-x64-v7.1.tgz

tar -xvf cudnn-9.0-linux-x64-v7.1.tgz && cd cuda
    sudo cp -P cuda/include/cudnn.h /usr/include
    sudo cp -P cuda/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
    sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*

TensorRT

# 从 tesnorRT 下载页面下载
    https://developer.nvidia.com/nvidia-tensorrt-5x-download

发表评论取消回复

表情：

评论列表（有 0 条评论，223人围观）

还没有评论，来说两句吧...

相关阅读

相关 CUDA和CUDNN安装和版本验证

> 提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档文章目录前言一、安装二、关键指标 1.驱动版本和CUDA版本对

灰太狼/ 2024年03月17日 14:20/ 0 赞/ 66 阅读

相关 PyTorch 安装 GPU版本(CUDA/cuDNN)

![20191009191333910.png][][日萌社][Link 1] [人工智能AI：Keras PyTorch MXNet TensorFlow Paddle

旧城等待，/ 2023年07月13日 03:36/ 0 赞/ 267 阅读

相关 Windows10安装CUDA和cuDNN

前言 > 我们都知道，CUDA和cuDNN是用来加速深度学习计算的，那么怎么在Windows上面安装CUDA以及cuDNN呢？准备阶段 > 1 自己电脑上应该有

客官°小女子只卖身不卖艺/ 2023年03月13日 02:07/ 0 赞/ 18 阅读

相关【全面】Centos7安装CUDA和cuDNN

本文讲解如何在centos7系统下下载安装CUDA和cuDNN,以下命令均在root权限下使用 1 确认电脑上已安装了GPU和显卡驱动 > .检查GPU是否安装及型

「爱情、让人受尽委屈。」/ 2023年03月12日 07:19/ 0 赞/ 16 阅读

相关 ubuntu安装CUDA、cudnn

1.查看GPU nvidia-smi 2.安装CUDA [https://developer.nvidia.com/cuda-downloads][https_d

ゝ一世哀愁。/ 2022年11月29日 12:26/ 0 赞/ 221 阅读

相关【TensorRT】ubuntu16.04+cuda10.1安装tensorrt6

参考：[https://zhuanlan.zhihu.com/p/64053177][https_zhuanlan.zhihu.com_p_64053177] 上面的参考写得

柔光的暖阳◎/ 2022年11月18日 04:55/ 0 赞/ 392 阅读

相关 cuda和cudnn下载地址

https://developer.nvidia.com/cuda-downloads? https://developer.nvidia.com/rdp/cu

悠悠/ 2022年09月03日 15:17/ 0 赞/ 189 阅读

相关 linux16.04安装cuda和cudnn, TensorRT

安装cuda wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x8

骑猪看日落/ 2022年05月11日 09:06/ 0 赞/ 224 阅读

相关 PyTorch下 CUDA 和 CuDNN 安装验证程序

装完gpu版的 pytorch，和给cuda装上cudnn后，不知道装对了没有，找了个验证程序验证一下 import torch 若正常则静默

深碍√TFBOYSˉ_/ 2022年04月17日 06:55/ 0 赞/ 257 阅读

相关 Ubuntu安装和卸载CUDA和CUDNN

目录文章目录 [目录][Link 1] [前言][Link 2] [安装显卡驱动][Link 3] [禁用nouveau

秒速五厘米/ 2021年12月03日 04:51/ 0 赞/ 668 阅读