栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

升级cuda+tenserrt+python(cuda11.3+tensorrt8.2.1+python3.7)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

升级cuda+tenserrt+python(cuda11.3+tensorrt8.2.1+python3.7)

之前装了cuda10.1+cudnn+tensorrt5.15+nvidia驱动,

现在想更新tesorrt到8.2,需要卸载已有的cuda+tensorrt,安装cuda11.3

由于我得用deb方式安装cuda,它会自动安装一个nvidia驱动,需要把现有的nvidia驱动卸载掉,否则两个驱动共存会导致开机无法进入ubuntu桌面。

1. 卸载原有nvidia相关软件

(1) 卸载cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

如果是deb方式安装的,还需要如下命令:

sudo rm -rf /usr/include/cudnn.h
sudo rm -rf /usr/lib/x86_64-linux-gnu/libcudnn*
sudo apt-get --purge remove "*libcudnn*"

(2) 卸载cuda10.1

sudo /usr/local/cuda-10.1/bin/cuda-uninstaller
cd /usr/local/
sudo rm -r cuda-10.1

如果是deb方式安装的cuda,用如下方式卸载cuda:

sudo apt-get --purge remove "*cublas*" "*cufft*" "*curand*" 
 "*cusolver*" "*cusparse*" "*npp*" "*nvjpeg*" "cuda*" "nsight*" 

(3)卸载tensorrt

sudo apt-get --purge remove nv-tenserrt*
sudo apt-get --purge remove "*tensorrt*"

(4)卸载nvidia驱动

sudo apt-get --purge remove "*nvidia*"

用如下命令查看是否卸载干净:

sudo dpkg --list | grep cuda

卸载干净的话,应该什么也找不到了,此时再开始  新的安装。

2. 安装

(1)安装cuda11.3

我选择deb方式安装,安装完成后查看一下当前nvidia驱动:

nvidia-smi

如果出现: Failed to initialize NVML: Driver/library version mismatch

重启一下就好。

我的nvidia驱动为465.19.01,cuda为11.3。

(2) 安装tensorrt+cudnn

安装8.2.5.1的tensorrt,用deb安装,期间会自动安装cudnn8.2.1。

(3) 升级python3.6——>python3.7

sudo add-apt-repository ppa:deadsnakes/ppa
sudo apt-get update
sudo apt-get install python3.7
#更换python3指向的软链接
sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.7 2
#切换python3指向
sudo update-alternatives --config python3

上面的命令作用是:安装了python3.7,并修改了python3命令的指向到python3.7。

原先的python3.6还在,并且可以通过python3.6来运行python3.6的程序。

安装在python3.6本地仓库的第三方库当然不能在运行python3.7时使用,需要为python3,7额外装第三方库。

此时pip3 install package 会向python3.7的仓库安装; apt install 方式安装的python包仍然是装到系统默认的python3.6。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/998548.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号