open-mmlab / mmaction2

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
https://mmaction2.readthedocs.io
Apache License 2.0
4.21k stars 1.23k forks source link

Whfat are optimal torch and cuda versions to build docker?[Docs] #2735

Open ankhafizov opened 11 months ago

ankhafizov commented 11 months ago

The doc issue

could you pliease give a table fro compatible torch-cuda versions to build docker when cuda version > 11.0. For example I have nvidia a40 dievice? which does not work recommended in github parameters:

ARG PYTORCH="1.8.1" ARG CUDA="10.2" ARG CUDNN="7"

Either it seems, that it does not work (or, probably I am doing something wrong. with

1.13.1-cuda11.6-cudnn8-devel

could you suggest any tested versions of torch, cuda and cudnn to start with?

Suggest a potential alternative/fix

No response