欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  网络运营

ubuntu安装显卡驱动和cuda教程

程序员文章站 2022-06-25 09:03:18
目录1. 卸载原始的驱动2. 下载新显卡驱动2.1 安装显卡驱动3 安装cuda查看nvcc -vcudatoolkit4. 安装cudnn5. 安装anaconda添加环境变量替换anaconda源...

写在最前面:

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

1. 卸载原始的驱动

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

2. 下载新显卡驱动

https://www.nvidia.cn/download/index.aspx?lang=cn

ubuntu安装显卡驱动和cuda教程

复制下载链接,在系统中用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 安装显卡驱动

ubuntu安装显卡驱动和cuda教程

ubuntu安装显卡驱动和cuda教程

ubuntu安装显卡驱动和cuda教程

3 安装cuda

官网链接

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

wget https://developer.nvidia.com/compute/cuda/10.0/prod/local_installers/cuda_10.0.130_410.48_linux
sudo sh cuda_10.0.130_410.48_linux

ubuntu安装显卡驱动和cuda教程

添加环境变量,将上图中的建议加到.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-11.4/bin
export ld_library_path=$ld_library_path:/usr/local/cuda-11.4/lib64
#使环境生效
source ~/.bashrc

查看nvcc -v

ubuntu安装显卡驱动和cuda教程

cudatoolkit

sudo apt install nvidia-cuda-toolkit

4. 安装cudnn

安装cudnn

https://developer.nvidia.com/rdp/cudnn-download

ubuntu安装显卡驱动和cuda教程

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

ubuntu安装显卡驱动和cuda教程

5. 安装anaconda

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

ubuntu安装显卡驱动和cuda教程

添加环境变量

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

ubuntu安装显卡驱动和cuda教程

替换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版本

ubuntu安装显卡驱动和cuda教程

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__

ubuntu安装显卡驱动和cuda教程

上述报错原因是cuda版本太高了,要选择10.1版本

ubuntu安装显卡驱动和cuda教程

上述报错原因是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

以上就是ubuntu安装显卡驱动和cuda教程的详细内容,更多关于ubuntu安装显卡驱动和cuda的资料请关注其它相关文章!