pytorch+cuda+onnx环境配置 去年捣鼓过cuda10.2,最近捣鼓yolov5从pytorch转换为onnx时出错了,捣鼓了半天。。。 这里提供一个稳定的环境配置: pytorch1.7.1+cuda11.0.03+cudnn8.0.2.39+onnx1.9+onxruntime(-gpu)1.8.0 ps:前提是支持cuda11.0,用命令nvidia-smi查看设备cuda最高支持什么版本。