ubuntu安装显卡驱动和cuda

望长安于日下

已于 2022-08-21 23:56:59 修改

阅读量1.7k

点赞数 2

分类专栏： linux 文章标签： ubuntu pytorch 深度学习

于 2021-09-01 00:07:24 首次发布

本文链接：https://blog.csdn.net/qq_26884501/article/details/120027034

版权

linux 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

文章目录

前言
1. 卸载原始的驱动
2. 下载新显卡驱动
- 2.1 安装显卡驱动
3 安装cuda
4. 安装cudnn
5. 安装anaconda

前言

最新的版本不一定是好的，合适的才是最好的，建议cuda10.1+cudnn7.6.5

1. 卸载原始的驱动

#查看安装的包
apt list --installed|grep -i nvidia
#卸载包
apt-get purge nvidia*

2. 下载新显卡驱动

能够查看历史版本：https://www.nvidia.cn/Download/Find.aspx?lang=cn
https://www.nvidia.cn/geforce/drivers/
https://www.nvidia.cn/Download/index.aspx?lang=cn
在这里插入图片描述

复制下载链接，在系统中用wget下载

#下载
wget https://cn.download.nvidia.cn/tesla/470.57.02/NVIDIA-Linux-x86_64-470.57.02.run

#安装
sudo sh NVIDIA-Linux-x86_64-470.57.02.run

2.1 安装显卡驱动

在这里插入图片描述

3 安装cuda

官网
https://developer.nvidia.com/cuda-toolkit-archive

选择cuda版本，要和驱动的cuda版本一致

wget wget https://developer.download.nvidia.cn/compute/cuda/10.1/secure/Prod/local_installers/cuda_10.1.105_418.39_linux.run
sudo sh cuda_10.1.105_418.39_linux.run

在这里插入图片描述

在这里插入图片描述
添加环境变量，将上图中的建议加到.bashrc文件中

Please make sure that

PATH includes /usr/local/cuda-11.4/bin
LD_LIBRARY_PATH includes /usr/local/cuda-11.4/lib64, or, add /usr/local/cuda-11.4/lib64 to /etc/ld.so.conf and run ldconfig as root

vim ~/.bashrc

#添加路径
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

#使环境生效
source ~/.bashrc

查看nvcc -V
在这里插入图片描述
cudatoolkit

sudo apt install nvidia-cuda-toolkit

4. 安装cudnn

安装cudnn
https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述

wget https://developer.download.nvidia.cn/compute/machine-learning/cudnn/secure/8.2.2/11.4_07062021/Ubuntu18_04-x64/libcudnn8_8.2.2.26-1%2Bcuda11.4_amd64.deb?aJLLhXbzztwE4iizwf68uvg1s73kk4KKBGqv6B0UkO9HhnOhOsGHlyo1Br5CWc0nAIJLmc6C5SkLYqbdQqdZBoAdcVQgBTmWKXJXigR7roUeXd0VIKUuM57UKWMp3BUQgr6SQ4kkGnRRtUJ5mJt

dpkg -i libcudnn8_8.2.2.26-1+cuda11.4_amd64.deb

在这里插入图片描述

5. 安装anaconda

wget https://mirror.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2021.05-Linux-x86_64.sh

在这里插入图片描述
添加环境变量

vim ~/.bashrc
export PATH="/usr/local/anaconda3/bin:$PATH"
source ~/.bashrc

在这里插入图片描述
替换anaconda源

"""更换清华conda源"""
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

查看tensorflow版本
在这里插入图片描述

pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

测试安装的tensorflow

import tensorflow as tf
print(tf.test.is_gpu_available())
tf.__version__
tf.__path__

在这里插入图片描述
上述报错原因是cuda版本太高了，要选择10.1版本

上述报错原因是cudnn版本太高了，要选择7.6.5版本

默认Python2调整为Python3

apt-get install python3.7

sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100
sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

sudo apt install python3-pip

安装cv2

pip install opencv-python
#如果出现问题
ImportError: libSM.so.6: cannot open shared object file: No such file or directory
sudo apt-get install -y python-qt4