python 查看显卡_ubuntu下安装显卡驱动、CUDA、CUDNN、pytorch过程

以ubuntu18.0.4为例，安装pytorch环境，会因为版本不一致导致程序无法正常运行，其中包括显卡驱动、cuda、cudnn、pytorch、python版本，但凡有一个不匹配，都会发生问题。

1、安装显卡驱动

显卡驱动处于最底层，被cuda所调用，一般来说，显卡驱动的版本高于cuda推荐最低版本即可。

去

显卡驱动下载官网​www.nvidia.cn

下载符合自己显卡的驱动，同时注意下载的版本号。

驱动安装完成后，可通过nvidia-smi查看显卡的具体版本。

2、安装cuda

去cuda官网下载cuda版本。这里需要注意，之前下载的显卡驱动和cuda版本有对应关系，详见官网说明

例如，如果需要安装cuda11.1以上，那么之前所安装的显卡驱动不能低于455.32的版本，可以高于455.32，但是绝对不能低。

cuda安装完成后，必须配置环境变量，否则程序将无法正确找到cuda所在的位置

vim ~/.bashrc

在末尾增加

export CUDA_HOME=/usr/loca