Ubuntu的各种环境配置

Ubuntu的各种环境

一、显卡驱动

nvidia的显卡驱动是必须安装的，不然显卡没法使用。如果后续需要配置cuda、tensorrt，我建议不单独安装驱动，因为deb方式安装cuda会自动安装驱动。
查看显卡驱动的方式：

方式一：查看当前系统所有nvidia驱动

sudo dpkg --list | grep nvidia-*

应该只有以ii开头的，并且驱动号唯一。如果存在rc开头，是有一些版本没有删除干净，可以使用命令

dpkg -l |grep ^rc|awk ‘{print $2}’ |sudo xargs dpkg -P

方式二：查看本次启动系统以为的版本

cat /proc/driver/nvidia/version

这是系统开机的时候，自动检查到的驱动版本

方式三：查看显卡状态

nvidia-smi

可以看到显卡使用情况，如果报错版本不匹配，多半是本次使用过程卸载并重装了驱动，或系统内核更新了，此时只要方式一查看的驱动版本唯一，即可重启解决问题，重启后方式二会更新为系统最新的驱动。
我是465.19.01

二、cuda

注意30系笔记本一定要cuda11版本以上，如果安装tensorRT推荐使用deb(local)方式安装cuda，这种情况下显卡驱动不需要单独安装。如果已经安装可以卸载掉

nvcc -V

或

cat /usr/local/cuda/version.json #也可能是txt

我是11.3.58

三、cudnn

也推荐使用deb方式安装，安装是否成功也可以通过例子测试。

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

我是8.2.1

四、anaconda

我是3.5.2.0

五、python,pip

python python2 python3都是需要软连接连接到你需要的python的。如果你安装了anaconda3，它会优先寻找anaconda3/bin里面的python软连接。
建议对应关系为：
python——ubuntu自带的python3.6
python2——ros自带的python2.7
python3——ubuntu自带的python3.6
如果使用anaconda的python，直接去env激活的环境就行了
pip 则需要使用
python -m pip …
python2 -m pip …
python3 -m pip …
安装给哪个python这样更清楚，为了加快速度，在pip install 时可以加上

-i https://pypi.tuna.tsinghua.edu.cn/simple

六、pytorch

如果是安装在anaconda，去pytorch官网找合适的即可，他的命令一定是cuda和torch相互对应的
如果是安装在系统，需要去pytorch官网找到对应你的cuda版本的pytorch，而且建议安装在python3中

python3
import torch
print(torch.__version__)

我是1.10.2

torch.cuda.is_available()

七、TensorRT

使用deb安装，按照官网说明安装即可。

cat /usr/include/x86_64-linux-gnu/NvInferVersion.h | grep NV_TENSORRT_MAJOR -A 3

我是8.2.5.1