ubuntu20.4安装NVIDIA+安装pytorch==1.8版本

安装NVIDI

安装之前先卸载已经存在的驱动

sudo apt-get remove --purge nvidia*

禁用nouveau

sudo gedit /etc/modprobe.d/blacklist.conf

添加到末尾：

blacklist nouveau
options nouveau modeset=0

选择适合版本进行下载

• 点开附加驱动里面的内容，选择适合的驱动进行应用更改就可以了
  

蓝屏界面perform MOK management处理

• 安装后从新启动电脑，然后会出现蓝屏界面：perform MOK management：进行选择
• 先进人enroll mok
• 再enroll mok界面中选择continue
• 进入enroll the mok界面中选择yes
• 输入密钥密码
• 之后到perform mok management 选择第一个reboot
• 上面的步骤成功后就装成功了

查看是否成功输入

nvidia-settings

在进行安装pytorch

进入pytorch官网：https://pytorch.org/
选择适合版本号进行选择下载

我安装的1.8版本

anaconda3下载
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

pip版下载
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

在浏览器上下载对应版本：
https://download.pytorch.org/whl/torch_stable.html

ubuntu安装pytorch会很慢使用以下方法：

注使用这个需要把 -c pytorch -c conda-forge去掉

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 

• 在终端输入：

sudo gedit ~/.condarc

• 编辑写入科大版：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
show_channel_urls: true

• 编辑写入其他版：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
show_channel_urls: true

下载很快

进行测试代码：

from __future__ import print_function  
import torch
x = torch.rand(5, 3)
print(x)





import torch
flag = torch.cuda.is_available()
print(flag)

ngpu= 1

完全卸载NAVIDIA驱动命令

sudo apt-get --purge remove nvidia*
sudo apt-get --purge remove "*nvidia*"
sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt autoremove

其它命令

# 查看NAVIDIA的型号
thanlon@thanlon:~$ lspci |grep -i nvidia
02:00.0 3D controller: NVIDIA Corporation GP108M [GeForce MX150] (rev a1)
# 查看NVIDIA驱动版本
sudo dpkg --list | grep nvidia-*
# 检查适合系统的NAVIDIA版本
thanlon@thanlon:~$ nvidia-detector 
nvidia-driver-440