mmaction2 icon indicating copy to clipboard operation
mmaction2 copied to clipboard

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

Open ankhafizov opened this issue 1 year ago • 0 comments

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

ankhafizov avatar Nov 01 '23 11:11 ankhafizov