机器视觉爱好者的博客

可能问题1：cuda 与 torch 版本不对应可能问题2：安装的torch是cpu版本，或者无cuda版本可能问题3：torch的关联包未安装解决方案：使用conda重新安装conda install pytorch torchvision cudatoolkit=10.1 -c pytorch10.1是cuda版本号conda 不仅会安装torch，还会安装关联包...

报错原因，cuda版本和驱动不对应。https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html上面是官网链接，显示最新的对应关系。（防止以后cuda升级后又要找）先贴出最新的：驱动下载链接：https://www.nvidia.com/Download/index.aspx?lang=en-usc...

先上图在上文字：Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.0/lib64:/usr...

直接上报错的图：前提条件：检查了cuda.is_available（）和CUDA_HOME为True解决方案：conda install -c pytorch pytorch-nightly cuda100 （我的cuda为cuda10.0）

解决方案：CUDA_DEVICE=0 python onnx_to_tensorrt.py

cuda和cudnn搭配 ubuntu16.04cuda9.0 cudnn7.0.5cuda10.0.130 cudnn 7.5.0查看 CUDA 版本：cat /usr/local/cuda/version.txt查看 CUDNN 版本：cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2...