JesseZhZhang

> Do a nvcc --version. Your CUDA needs to be 11.7 If not, then conda install -c "nvidia/label/cuda-11.7.0" cuda-toolkit and conda install -c "nvidia/label/cuda-11.7.0" cuda > > But, I have...

perhaps you can ask gpt to generate a training demo

For me, it is a problem with internet connection. I used` HF_ENDPOINT=https://hf-mirror.com` and fixed the problem (of course, only if you are in mainland China)

> @JesseZZZZZ > > try > > ```python > import torch > from vit_pytorch.simple_flash_attn_vit_3d import SimpleViT > > v = SimpleViT( > image_size = 128, # image size > frames...