目录

一、安装显卡驱动

1.查看显卡

2.查看可安装的显卡驱动版本

3.重启。

4.驱动检测命令:

二、卸载显卡驱动

三、安装CUDA

2.下载的CUDA版本

3.检测是否安装成功        输入命令:nvcc -V,弹出如下则表示成功

四、卸载CUDA

       1. 卸载命令如下:

        2.切换到CUDA所在目录:

五、CUDNN 的安装

1.下载所有的deb文件

2.安装deb文件

3.测试CUDNN


一、安装显卡驱动

1.查看显卡

lspci -k | grep -A 2 -i "VGA"

2.查看可安装的显卡驱动版本

ubuntu-drivers devices

我选择的是 nvidia-driver-510 - third-party non-free recommended, 对应的安装命令为:sudo apt install nvidia-driver-510。

3.重启。

4.驱动检测命令:

nvidia-smi

二、卸载显卡驱动

sudo apt-get --purge remove nvidia*
sudo apt autoremove

To remove CUDA Toolkit:
$ sudo apt-get --purge remove "*cublas*" "cuda*"

To remove NVIDIA Drivers:
$ sudo apt-get --purge remove "*nvidia*"

三、安装CUDA

1.首先,查看不同CUDA10.2所需的驱动版本要求 https://docs.nvidia.com/cuda/archive/,点击对应的版本,在此,选择10.2版本。如下图所示,点击左上角的 Release Notes ,可以看到cuda10.2再Ubuntu环境下显卡驱动版本>=440。

2.下载的CUDA版本

进入官网 CUDA 10.2,选择要下载的CUDA版本:

 方式2和3中的deb方式,由于网络原因一直下载不下来,导致安装失败。故直接根据runfile(local)方式安装。

按照官网教程:

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

sudo sh cuda_10.2.89_440.33.01_linux.run

注意:在运行 run文件时,会出现三次提示,在第三次提示时取消 Drives 的勾选(因为显卡驱动已经安装),否则,cuda10.2默认的显卡驱动版本和自己电脑已安装版本冲突会导致cuda安装失败。

 安装失败如下:

安装成功后会出现如下界面:

3.检测是否安装成功
        输入命令:nvcc -V,弹出如下则表示成功

四、卸载CUDA

       1. 卸载命令如下:

    sudo apt-get remove cuda
    sudo apt autoremove 
    sudo apt-get remove cuda*

        2.切换到CUDA所在目录:

cd /usr/local/

        删除要删除的CUDA目录:

sudo rm -r cuda-**.**.**

为确保cuda删除干净,可通过下述命令行验证。

sudo dpkg -l |grep cuda

若存在,则用下述命令行逐一删除即可。

sudo dpkg -P cuda-**.**.**

五、CUDNN 的安装

1.下载所有的deb文件

CUDNN_version 选择自己所需的版本下载所有的deb文件,所有是指:Runtime Library,Developer Library和Code Samples的deb文件,如下图所示,下载第5、6、7个文件。

2.安装deb文件

sudo dpkg -i libcudnn8_8.0.2.39-1+cuda10.2_amd64.deb 

sudo dpkg -i libcudnn8-dev_8.0.2.39-1+cuda10.2_amd64.deb 

sudo dpkg -i libcudnn8-doc_8.0.2.39-1+cuda10.2_amd64.deb 

3.测试CUDNN

cp -r /usr/src/cudnn_samples_v8/ $HOME

cd  ~/cudnn_samples_v8/mnistCUDNN

sudo make clean 
sudo make

sudo ./mnistCUDNN

最终,出现下述界面,即安装成功。

完结,撒花~

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐