服务器新装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指南