mmaction2
mmaction2 copied to clipboard
Whfat are optimal torch and cuda versions to build docker?[Docs]
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