https://dl.fbaipublicfiles.com/detectron2/wheels/cu117/torch2.0/index.html怎么处理
时间: 2023-09-05 21:08:02 浏览: 298
这个链接是 Detectron2 在 CUDA 11.1 和 PyTorch 1.8 环境下的预编译二进制包,如果你的系统环境符合这些要求,可以直接使用这个链接安装 Detectron2,命令如下:
```
pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu117/torch1.8/index.html
```
如果你的系统环境不符合这些要求,需要根据自己的系统环境重新编译 Detectron2,可以参考 Detectron2 官方文档中的 [安装文档](https://detectron2.readthedocs.io/en/latest/tutorials/install.html) 进行安装。
相关问题
python -m pip install detectron2==0.4 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.7/index.html
这是一个使用 pip 安装 detectron2 版本 0.4 的命令,并且指定了 CUDA 版本为 10.2,PyTorch 版本为 1.7。其中 `-f` 参数指定了附加的 wheel 文件源,即 Facebook AI Research 提供的 detectron2 wheel 文件的下载地址。
conda create -n prompt python=3.7 conda activate prompt pip install -q tensorflow # specifying tfds versions is important to reproduce our results pip install tfds-nightly==4.4.0.dev202201080107 pip install opencv-python pip install tensorflow-addons pip install mock conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu110/torch1.7/index.html pip install opencv-python conda install tqdm pandas matplotlib seaborn scikit-learn scipy simplejson termcolor conda install -c iopath iopath # for transformers pip install timm==0.4.12 pip install ml-collections # Optional: for slurm jobs pip install submitit -U pip install slurm_gpustat
这段代码是用来创建一个名为 "prompt" 的 Conda 环境,并安装一些常用的 Python 库和机器学习框架,例如 TensorFlow、PyTorch、Detectron2、Transformers 等。其中,指定了 TensorFlow Datasets 的版本号为 4.4.0.dev202201080107,这是为了能够复现作者的实验结果。另外,还安装了一些用于辅助开发的库,如 opencv-python、tqdm、pandas、matplotlib、seaborn、scikit-learn、scipy、simplejson、termcolor 等。如果要在 Slurm 集群上运行分布式训练任务,可以安装 submitit 和 slurm_gpustat。
阅读全文