Does this code run only on V100, which does not support 3090?

linjieli222 / HERO

Research code for EMNLP 2020 paper "HERO: Hierarchical Encoder for Video+Language Omni-representation Pre-training"

https://arxiv.org/abs/2005.00200

MIT License

230 stars 34 forks source link

Does this code run only on V100, which does not support 3090? #45

Open zhangxiaoxiao-tech opened 2 years ago

zhangxiaoxiao-tech commented 2 years ago

The following error occurs when running on 3090：

“WARNING: Detected NVIDIA GeForce RTX 3090 GPU, which is not yet supported in this version of the container ERROR: No supported GPU(s) detected to run this container”

linjieli222 commented 2 years ago

Our docker is built on V100 GPUs and inherited from nvidia built pytorch docker: pytorch:19.10-py3 from here. If you wish to support on your own GPU version, you may need to rebuilt the docker image with compatible pytorch dockers from nvidia.

linjieli222 commented 2 years ago

If pytorch:19.10-py3 is compatible with RTX 3090, you can simply run docker build ./ -t <your_docker_image_name> to rebuild the same enviroments to the released docker image.