以下安装环境均为win10的Anaconda。
1.在conda环境中离线安装对应版本的pytorch和torchvision
conda install --offline pytorch-1.6.0-py3.7_cuda102_cudnn7_0.tar.bz2 #文件全称
conda install --offline torchvision-0.7.0-py37_cu102.tar.bz2 #文件全称
参考:win10下conda安装pytorch-gpu版本(超详细),完美解决镜像源下载慢问题!_皮皮鲁与鲁西西�的博客-CSDN博客
2.在线安装除torch和torchvision之外的包,比如onnx,onnxruntime,tensorrt等
conda install torchaudio cudatoolkit=11.3
参考:用anaconda下载安装pytorch1.8.1+cudatoolkit11.1_舍得叔叔的博客-CSDN博客_cudatoolkit=11.1
3.安装onnx-simp。conda环境下直接pip安装
pip3 install -U pip && pip3 install onnx-simplifier
参考:GitHub - daquexian/onnx-simplifier: Simplify your onnx model
4.安装lmdb。conda环境下直接安装
conda install -c conda-forge python-lmdb