PyCenterNet2
zzzzzzz286972發表於2024-11-26
- V100
conda create -n myenv python=3.8
conda activate myenv
conda install pytorch=1.10 cudatoolkit=11.3 -c pytorch
pip install -U numpy==1.23.5
pip install mmcv-full==1.4.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10/index.html
python setup.py develop
pip install torchvision==0.11.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install huggingface_hub safetensors -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
參考連結:https://blog.csdn.net/qq_36622589/article/details/124355564