Quick fix for Gitlab runner memory issue
Fix torch version to 1.12.1 and cudatoolkit to 11.3 to avoid memory issue when creating the docker container.
Related to #104 (closed)
Fix torch version to 1.12.1 and cudatoolkit to 11.3 to avoid memory issue when creating the docker container.
Related to #104 (closed)