In our experiments, we use CUDA11.0 and pytorch1.7.1 as our basic docker environment. Does it suitable for the final verification stage?Posted by: rhythmblue @ Jan. 11, 2021, 9:49 a.m.
Thanks for asking. We recommend CUDA version 10.1 or lower.
However, the machine that will be used at the code verification stage has CUDA version 10.1 installed but Driver Version: 450.80.02, which should handle CUDA Version: 11.0 with no problem.