Pytracking installation on Linux

最新推荐文章于 2022-11-16 22:12:29 发布

原创

最新推荐文章于 2022-11-16 22:12:29 发布 · 419 阅读

1 ·

CC 4.0 BY-SA版权

本文档提供了在Ubuntu 18.04系统上安装PyTracking所需依赖的详细步骤。首先需要安装Conda和Python 3.7，然后安装支持CUDA的PyTorch。接着安装matplotlib、pandas等库，以及coco和lvis工具包。还需要安装ninja-build、spatial-correlation-sampler、jpeg4py，并设置环境。最后下载预训练网络。

Linux（Ubuntu ）Installation

原文github
This document contains detailed instructions for installing the necessary dependencies for PyTracking. The instrustions have been tested on an Ubuntu 18.04 system. We recommend using the install script if you have not already tried that.

Requirements

Conda installation with Python 3.7. If not already installed, install from https://www.anaconda.com/distribution/.
Nvidia GPU.

Step-by-step instructions

Create and activate a conda environment(使用克隆功能会更快更好地创建一个新环境)

conda create --name pytracking python=3.7
conda activate pytracking

Install PyTorch
Install PyTorch with cuda10.(推荐使用cuda 11 for RTX3090 GPU)

#conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c conda-forge
conda install pytorch torchvision torchaudio cudatoolkit=11.1

最低0.47元/天解锁文章