qq_64997742的博客

目录 1.卸载旧驱动2. 禁用nouveau驱动3. 安装最后出现nvidia-drm错误1）开启了UEFI secure boot，导致第三方模块没有被合法签名，所以无法校验成功并加载2）其他安装，导致系统在配置文件中禁用了nvidia-drm 模块不起作用4. 安装Nvidia驱动5.安装显卡驱动6. 我这里选择recommended的驱动版本进行安装：7. 安装CUDA1. 在官网（CUDA Toolkit Archive | NVIDIA Developer）下载cuda，版本根据自己需求选择即可，

确保正确导出onnx文件需要注意以下几个方面：1：任何用到shape/size返回值的参数时，例如tensor.view(tensor.size(0), -1)这种操作，避免直接使用tensor.size()的返回值，而是加上int强转，例如tensor.view((tensor.size(0)), -1)，断开跟踪。2：nn.Upsample或者nn.functional.interpolate函数，使用scale_factor指定倍率，而不是用size参数指定。