cuda10.2 and 3060 do not match

[researchmm]

cuda10.2 and 3060 do not match

[MasterBin-IIAU]

@Richard-mei Hi, I think the CUDA version doesn't have an influence on this project. So you can choose the compatible CUDA and the PyTorch version based on your GPU. For example, you can create a new conda environment then run the install_pytorch17.sh but replace the original command conda install -y pytorch==1.7.0 torchvision==0.8.1 cudatoolkit=10.2 -c pytorch with conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch.

[Richard-mei]

I have tried pytorch==1.7.0 torchvision==0.8.1 torchaudio==0.7.0 cudatoolkit=11.0, but as I said above, an error about tensorboardX occured.

[MasterBin-IIAU]

We haven't tested our code on a 3060 GPU. However, if you only want to run demos or test the STARK tracker, you can just comment out all lines about tensorboard.

[Richard-mei]

Thank you. I replaced with conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge and it works.