MeMotr多目标追踪环境配置

蜜汁小烤肉

已于 2024-12-13 23:18:12 修改

阅读量259

点赞数 1

CC 4.0 BY-SA版权

文章标签： python 深度学习目标跟踪目标检测

于 2024-12-13 16:16:48 首次发布

本文链接：https://blog.youkuaiyun.com/dddsnnf/article/details/144441503

显卡：4090

cuda：12.1(我的cuda使用11.8版本不行，提示cuda is not avaliable)

检查cuda设别是否可用

python
import torch
print(torch.cuda.is_available())

python：3.10

pytorch：

conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia

conda install -c conda-forge lap

conda install matplotlib
conda install pyyaml
conda install tqdm
conda install tensorboard

conda install cython

conda install seaborn

conda install dill

pip install scipy -i https://pypi.tuna.tsinghua.edu.cn/simple/

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple/