Deepin20.6 RTX3080 安装显卡驱动510.60.02、CUDA11.6、PyTorch1.11
0 前言
近期有个目标检测的任务，需要使用CUDA训练网络模型，使用的设备是RTX3080，在Deepin操作系统下，安装了510.60.02版本的驱动，简单记录下操作步骤，以备不时之需
1 准备工作
网上有很多帖子是通过英伟达官网下载.run格式的驱动来手工安装的，但是我尝试了很多版本，虽然能装上，但是装完之后无法开机，因此，在开始安装显卡驱动可CUDA套件前，建议先备份一下当前的操作系统
2 清理环境
卸载当前已经安装的英伟达驱动
sudo apt autoremove nvidia*
屏蔽开源驱动
sudo vi /etc/modprobe.d/blacklist.conf
按i进入vi的编辑模式，添加以下几行内容，然后ESC进入命令模式wq进行保存
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
如果开机卡黑屏，或者开机有如下警告
nvidiafb: cannot request PCI regions
可以在上述文件中多添加一行
blacklist nvidiafb
文件保存成功后，执行下面的命令，完成内核模块的屏蔽
sudo update-initramfs -u
重启机器，使用下面的命令检查，没有输出就屏蔽成功了
lsmod | grep nouveau
3 安装显卡驱动
不知道是Deepin的系统原因还是我台式机的原因，英伟达官网.run形式的驱动文件我装不成功，装上以后开机卡黑屏，然后在Deepin官方仓库里找到了可以直接使用的driver…
sudo apt search nvidia-driver
...
...
glx-alternative-nvidia/未知,now 1.2.0-1-1+eagle amd64 [已安装，自动]
allows the selection of NVIDIA as GLX provider
libegl-nvidia0/未知,now 510.60.02-1 amd64 [已安装，自动]
NVIDIA binary EGL library
...
...
喜出望外，直接install，一把梭哈，成功！
sudo apt install nvidia-driver
nvidia-smi
4 安装PyTorch & CudaToolKit
我安装的510.60.02版本自带的CUDA版本是11.6，在pytorch官网可以看到，当前最新PyTorch Stable版本是1.11.0，支持CUDA 11.3，CUDA11系列向下兼容低版本，选择自己的安装方式安装PyTorch 1.11.0即可，下方会有安装命令，贴终端跑一下完事，（记得更换conda或者pip到国内源，快一些）
conda安装
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip安装
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
5 检查
终端输入python or python3，进入python交互环境，检查下PyTorch和CUDA的版本，以及CUDA是否可用，至此，Deepin20.6+RTX3080安装显卡驱动、PyTorch、CUDA工作完成
>>> import torch
>>> torch.__version__
'1.11.0+cu113'
>>> torch.cuda.is_available()
True
>>> torch.version.cuda
'11.3'
>>>
Original: https://blog.csdn.net/qq_18625805/article/details/125223467
Author: first_Dance
Title: Deepin20.6 RTX3080 安装显卡驱动510.60.02、CUDA11.6、PyTorch1.11
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/710025/
转载文章受原作者版权保护。转载请注明原作者出处！