cuda安装

①cuda彻底删除

cd /usr/local/
sudo rm -r cuda*
sudo apt-get autoremove --purge cuda   # 彻底删除（把nvidia驱动也删了）
sudo dpkg -l | grep cuda   # 检查是不是没了

 ②安装驱动

　　查看命令： nvidia-smi  两种方式安装（多试几次）

• sudo ubuntu-drivers autoinstall
• ubuntu-drivers devices查看显卡和推荐驱动安装版本  sudo apt install nvidia-driver-495

③cuda .run文件的安装方式

　　进入链接(cuda旧版本) 选择进入11.1, 执行wget https://***.run、 chmod +x ***.run、 sudo sh ./***.run

   

 看到图(3)我还以为报错了, 接下来就是设置环境变量了

$ vi ~/.bashrc
export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
$ source ~/.bashrc
$ nvcc -V

④cudnn安装

 　　进入链接(cudnn所有版本)选择对应的版本与设备的硬件，下载***.tgz文件。

　　tar -zxf ***.tgz、  ls cuda（解压生成的目录），  然后就是配置了

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

sudo ldconfig /usr/local/cuda/lib64    # 链接一下，以防止找不到 

⑤测试cuda和cudnn

import torch
from torch.backends import cudnn

torch.cuda.is_available()        # 返回True 
cudnn.is_available()  　　　　    # 返回True
x =  torch.tensor(1.0)
cudnn.is_acceptable(x.cuda())    # 返回True

cuda与torch的版本对应关系（torch==1.8.0+cu111、torch==1.8.0、torch=1.8.0+cpu三种类型视硬件而定）

 

 2022-01-23　13:29:23