一文解决ubuntu20上3060显卡驱动,cuda,cudnn,torch的安装和卸载
文章目录
-
- 前言
-
- 1、显卡驱动安装
- 2、torch - torchvision - python 版本对应关系（==更推荐后面第7部分的一键安装，简单省事==）
- 3、cuda - torch-python 版本对应关系
- 4、确定cuda的版本为11.1，cudnn的版本为8.9.0，进入以下链接进行下载
- 5、cuda的安装
- 6、cudnn的安装
- 7、torch和torchvison的安装
- 8、torch和torchvison的卸载
- 9、cuda和cudnn的卸载
- 10、win10安装Ubuntu系统
- 11、ROS安装
- 总结
- 参考文章
前言
本文主要介绍在ubuntu20下根据自己的python版本来确定torch版本，从而确定cuda和cudnn版本完成安装（注意，本文可能会同时看到cuda11和cuda10，因为是第一次安装cuda10在3060显卡跑不起来就换成cuda11了），本文版本为python-3.8，torch-1.8.0，torchvision-0.9.0，cuda-11.1.1，cudnn-8.9.0。
1、显卡驱动安装
（1）通过以下命令查看匹配的驱动：
ubuntu-drivers devices
从上图看到recommended 470，故安装470版本的驱动
（2）驱动安装方法1，
sudo ubuntu-drivers autoinstall
（3） 推荐 驱动安装方法2，进入官网下载对应的版本
（4）安装相关依赖,
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
（5） 禁用系统默认显卡驱动，打开文件
sudo gedit /etc/modprobe.d/blacklist.conf
在文件末尾输入，
blacklist nouveau
options nouveau modeset=0
保存(ctrl+s)后手动更新；
sudo update-initramfs -u
重启后输入以下命令，无输出则禁用成功，
lsmod | grep nouveau
（6）配置环境变量，gedit命令打开配置文件：
sudo gedit ~/.bashrc
在文件末尾输入，
export LD_LIBRARY_PATH=/usr/lib/x86_64-linux