[Bug]: For `cv` problem template, removing `cpuonly` package doesn't enable GPU; CPU and GPU images does not work as intended

[Syakyr]

Problem Domain

• Docker
• Python

OS/Platform(s) Used

• None

Problem Brief

The cv problem template is meant to be used as an example for AIAP MLOps Week. And within the week, GPUs are not given to be used, so this issue would fly under the radar during the session. However, if GPUs are used for the guide, just removing the cpuonly package doesn't make that happen as the Pytorch version installed uses a build that only uses CPU even if GPU and and CUDA are available within the container. Thus, the current Dockerfile used to create GPU images is redundant and bloated since the Pytorch package wouldn't use the GPUs attached to the container.

Steps to Reproduce

• Attach GPUs to the container created using the *-gpu.Dockerfile with the cv problem template
• Run Python/iPython within the container
• Run import torch; torch.cuda.is_available()

Expected Result

true

Actual Result

false

[Syakyr]

To mitigate the issue, a separate yaml file is created to make the GPU image variant.