Comments (4)
@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
.
from stark.
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.
from stark.
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.
from stark.
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.
from stark.
Related Issues (20)
- the checkpoint file of stark-st1 link has expired HOT 2
- Where to download STARKST_ep0500.pth.tar ? HOT 2
- How to create the data folder with all the datasets on it?
- only one dataset HOT 2
- forward_backbone() missing 1 required positional argument: 'mask'
- the AUC of UAV123?
- the anno file of NFS
- Hyperparameter tuning
- StarkLightning的训练日志公布了吗?
- 想问下作者代码中的知识蒸馏部分是基于什么实现的? HOT 1
- Network outputs is NAN! HOT 1
- Questions about the training dataset
- Problem of Training for Stark-Lighting-X-trt HOT 1
- 目标走出视野后仍在跟踪
- 1
- How to get score or confidence from onnx
- 模型下不下来
- Some questions about AR(Alpha Refine).
- Qualifications
- checkpoint HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from stark.