驱动安装

为防止驱动安装后没…不知道咋弄的ubuntu20.04电脑的cuda驱动丢了无奈需装PyTorch环境只有CUDA11.0以上版本才支持Ubuntu20.04所以安装了CUDA11.0、cuDNN8.0.5

驱动安装

为防止驱动安装后没办法进入登陆界面这没有必要进行单独下载安装只要在操作界面进行点击安装就行。

具体如下在软件更新-附加驱动点击应用就行

驱动安装

nvidia-smi就可以看到这些信息如果命令可用说明驱动安装完成。

cuda安装

首先在Nvidia官网下载适合自己机器的CUDA版本官网下载本次使用

runfile

的方式安装用deb格式安装的话会在安装过程中替换掉已经安装好的显卡驱动如图所示

1.驱动下载

小版本可以暂时不用太关注。

接下来根据自己的电脑系统进行选择就行具体如下

2.执行安装

https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run

sudo

cuda_11.0.3_450.51.06_linux.run点击continue继续执行并将

CUDA

PATH/usr/local/cuda-11.0/bin${PATH::${PATH}}

export

LD_LIBRARY_PATH/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH::${LD_LIBRARY_PATH}}source

~/.bashrc

/usr/local/cuda/include/cudnn.h

/usr/local/cuda/lib64/libcudnn*

sudo

libcudnn8_8.0.5.39-1cuda11.0_amd64.deb

sudo

libcudnn8-dev_8.0.5.39-1cuda11.0_amd64.deb

sudo

libcudnn8-samples_8.0.5.39-1cuda11.0_amd64.deb

3.测试