博客

【初探DETR】安装环境

  • Post author:
  • Post category:其他


windows下 需要环境 

cython
git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI&egg=pycocotools
torch>=1.5.0
torchvision>=0.6.0
git+https://github.com/cocodataset/panopticapi.git#egg=panopticapi
scipy
onnx
onnxruntime

1.cython

pip install Cython

2.torch torchvision

我是在Windows下，安装的cuda10.2，大家可以自行参考，或者上

pytorch官网

查看,还是很好用的。找自己指定版本，或者自己修改到指定版本。 

pip install torch==1.9.1+cu102 torchvision==0.10.1+cu102 torchaudio -f https://download.pytorch.org/whl/cu102/torch_stable.html


版权声明：本文为kuoyang原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。