安装显卡驱动

在这里介绍一种有用、快速的安装方法:

1.进入linux系统,如图,找到应用”软件和更新“(请注意是软件和更新,不是软件更新器)

打开它,点击“附加驱动”

然后点击“应用更改”,输入密码认证后,等待进度条完成后,重启系统,即可完成安装。

开机后输入“nvidia-smi”,查看是否有驱动相关信息,如果有即安装成功

注意:

我在安装的时候前面按步骤都正常,开机之后输入nvidia-smi仍然报错 No devices were found,这个时候我重新选了附加驱动中别的选项,重新安装了一次后成功

安装Anaconda

安装Cuda

通过这个网址寻找适合自己的cuda版本

https://developer.nvidia.com/cuda-11-7-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local

运行下面命令进行安装

sudo sh 文件名

在“driver”处点击回车(取消选中),按↓移动到“install”,点击“回车”

安装Cudnn

通过这个链接Log in | NVIDIA Developer下载cudnn

可以根据自己的情况下载版本,我下载了此版本Local Installer for Ubuntu20.04 x86_64 (Deb)。

首先在cudnn所在的文件夹打开终端,然后输入下面代码进行解压

依次执行以下语句进行安装:

此处需要特别注意cuda*.*不一定是你之前安装的版本号,因此你需要到解压的*.deb文件夹中去看一看到底是cuda的版本号

sudo dpkg -i llibcudnn8_8.0.5.39-1+cuda11.0_amd64deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.0_amd64.deb

通过以下语句进行测试:

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd  ~/cudnn_samples_v8/mnistCUDNN
sudo make clean 
sudo make
sudo ./mnistCUDNN

安装pytorch

通过这个链接寻找适合自己的pytorch版本

https://pytorch.org/get-started/previous-versions/
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐