Hi there, Im wondering which cuda version used in final docker, need to install torch 1.10.0 or higher.
Posted by: xwj @ May 7, 2022, 6:44 a.m.Our docker has CUDA Version 10.1 (default) and 10.0 installed. If these versions does not meet your requirements, you can install a new version of CUDA and modify the .bashrc to indicate the valid version, e.g. by adding,
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
You can try and test using our docker and indicate the installing and configuration commands in your submitted code package, e.g. in a readme file.
Posted by: bob_peng @ May 7, 2022, 9:22 a.m.