当前位置: 代码迷 >> 综合 >> ubuntu20.02重装:向日葵+openssh-server+cuda+cudnn+tensorrt环境的搭建
  详细解决方案

ubuntu20.02重装:向日葵+openssh-server+cuda+cudnn+tensorrt环境的搭建

热度:67   发布时间:2024-01-21 02:08:25.0

服务器新装Ubuntu20.0系统之后需要一系列的操作:

1.安装向日葵

请移步:安装向日葵攻略

2.安装openssh-server(远程ssh需要)

#安装openssh-server
sudo apt update
sudo apt install openssh-server
#启动 openssh-server
sudo /etc/init.d/ssh start
# 或
sudo systemctl start ssh.service

3.安装miniconda3

安装Miniconda地址在:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/
我选了:Miniconda3-py38_4.9.2-Linux-x86_64.sh

安装指令:
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py38_4.9.2-Linux-x86_64.sh
chmod u+x Miniconda3-py38_4.9.2-Linux-x86_64.sh
./Miniconda3-py38_4.9.2-Linux-x86_64.sh

安装完成后,重启终端

#创建虚拟环境:
conda create -n tensorrt python=3.7
#激活虚拟环境
conda activate tensorrt

4.安装cuda

查看显卡驱动版本
在这里插入图片描述
驱动版本510 ,去官网:https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda,选择你的ubuntu版本即可,我的为20.04的版本即可
在这里插入图片描述
最好采用runfile方式
在这里插入图片描述
安装cuda
输入命令进行安装

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
sudo sh cuda_11.6.0_510.39.01_linux.run

具体过程参考:https://blog.csdn.net/h3c4lenovo/article/details/119003405
安装成功:nvcc -V
在这里插入图片描述

5.安装cudnn

官网:https://developer.nvidia.com/cudnn
下载对应cuda版本的cudnn
在这里插入图片描述
下载好cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz,进行解压tar -xvf cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz
官方指南:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html添加链接描述

sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.6/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.6/lib64
sudo chmod a+r /usr/local/cuda-11.6/include/cudnn*.h 
sudo chmod a+r /usr/local/cuda-11.6/lib64/libcudnn*

判断cudnn成功与否?

cat /usr/local/cuda-11.6/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

代码中的cuda-11.6改成对应的
参考:https://zhuanlan.zhihu.com/p/72298520

6.安装Tensorrt

安装tensorrt指南

  相关解决方案