卸载cuda:
cd /usr/local/cuda/bin/
sudo ./uninstall_cuda_10.0.pl
cd /usr/local/
sudo rm -r cuda-10.0/
卸载驱动:
cd /usr/bin/
./nvidia-uninstall
或者
sudo apt-get --purge remove nvidia*
==============================================================
安装Tensorrt:
sudo dpkg -i nv-tensorrt-repo-ubuntu1x04-cudax.x-trt5.x.x.x-ga-yyyymmdd_1-1_amd64.deb是把tensorrt安装在home下面,但是在pytorch虚拟环境里面没办法import tensorrt。
把tensorrt安装在pytorch虚拟环境下面:
登录NVIDIA官网NVIDIA TensorRT | NVIDIA Developer,然后下载tar.gz压缩包,并解压到home目录,再在虚拟环境里面安装
(pytorch-yolov5) yinshaohui@sysadmin-System-Product-Name:~/TensorRT-7.0.0.11/python$ pip install tensorrt-7.0.0.11-cp37-none-linux_x86_64.whl
参考: