硬件环境:
1、显卡:2080Ti
2、CUDA环境:cuda11.1
虚拟环境:
1、python:3.8
2、pytorch:1.10.1+cu111
3、torchvision :0.11.2+cu111
附上pytorch、torchvision对应cuda版本安装命令(以cuda11.1为例):
官方链接:Previous PyTorch Versions | PyTorch
# CUDA 11.1
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
切记:安装环境条件一定要按照detectron2官方来:
-
Linux 或 macOS with Py