【Ubuntu16.04 NVIDIA驱动，cuda11.3，cudnn8.2.1，对应torch安装】_cuda11.3 对应英伟达的显卡驱动

Ubuntu16.04 NVIDIA驱动，cuda11.3，cudnn8.2.1，对应torch安装

NVIDIA驱动安装：

提示：先输入nvidia-smi 查看是否有驱动，如果出现nvidia-smi: command not found 说明没有驱动，再开始下面安装，如果有但是想升级可以先把之前的驱动删除，再重新安装

这个是我升级以后的驱动器版本
进入正题
首先查看显卡型号：

 lspci | grep -i nvidia
1

可以看到这里显卡是[Tesla P100 PCIe 16GB] ，现在去 NVIDIA官网链接下载对应的驱动器

下载好后放入/home/ubuntu 目录下，

cd /home/ubuntu       #进入文件夹
ls    #可以在文件夹下看到下载的安装包  NVIDIA-Linux-x86_64-470.161.03.run
sh NVIDIA-Linux-x86_64-470.161.03.run  #开始安装
1
2
3

安装就一路接受就可以了，注意在ubuntu用户下，root下会出现问题
再次nvidia-smi，就会出现最顶图片，表示驱动安装完成

cuda11.3安装

去官网查找对应版本

wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
sudo sh cuda_11.3.1_465.19.01_linux.run
1
2

输入accept


enter取消 ，选中Install 回车

vim ~/.bashrc #配置路径
#写在最后
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda 
#最后
source ~/.bashrc
#查看
nvcc -V
1
2
3
4
5
6
7
8
9

可以看到已经有了

cudnn8.2.1安装：

去官网查找对应版本

下载需要注册，也不是很麻烦可以注册一下

tar -xvzf cudnn-11.3-linux-x64-v8.2.1.32.tgz   #解压安装包
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
1
2
3
4

对应torch安装：

其他版本可以去官网找，

 pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
1

写在最后，环境配置还是得多百度，毕竟问题多且杂，小菜鸟的心得
这个文章也是我不断踩坑慢慢摸索出来的，（感恩的心）