update instructions + slurm

This commit is contained in:
√(noham)²
2024-07-22 20:32:20 +02:00
parent 15b4fecdd0
commit db075fb427
5 changed files with 1811 additions and 0 deletions

View File

@@ -16,12 +16,28 @@ mv Yolov7-tracker yolov7-tracker
cd yolov7-tracker
git checkout v2 # change to v2 branch !!
```
Change np.int to np.int64 in :
utils/datasets.py
utils/general.py
Fix for GPU in [utils/loss.py](../yolov7-tracker-example/utils/loss.py.bak):
```python
from_which_layer.append((torch.ones(size=(len(b),)) * i).to('cuda')) ### SF #### (lines 686)
fg_mask_inboxes = fg_mask_inboxes.to(torch.device('cuda')) ### SF #### (lines 758 & 5164 & 1568)
```
Set up env :
```bash
conda create -n yolov7 python=3.9 pytorch=1.12
conda activate yolov7
pip3 install numpy scipy matplotlib cython pandas cuda-python
```
```
conda env : /home/noham.rivoirard/.conda/envs/yolov7
```
<!-- ## ERROR: Could not find a version that satisfies the requirement cuda-python (from versions: none) Requires-Python >=3.10 -->
```bash
pip3 install -r requirements.txt