Update DockerfileTGWUI

[cbluth95]

Update to Cuda 12.1 to support PyTorch

Description

Related Issues

Solution and Design

Test Plan

Type of change

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)
• Docs update

Checklist

• My pull request is atomic and focuses on a single change.
• I have read the contributing guide and my code conforms to the guidelines.
• I have documented my changes clearly and comprehensively.
• I have added the required tests.

[CLAassistant]

All committers have signed the CLA.

[joshuacox]

When trying the patch-1 PR I get this error:

Error response from daemon: failed to create task for container: failed to create shim task: OCI runtime create failed: runc create failed: unable to start container process: error during container init: error running hook #0: error running hook: exit status 1, stdout: , stderr: Auto-detected mode as 'legacy'
nvidia-container-cli: device error: 1: unknown device: unknown

On this same machine I can run gpu containers just fine:

docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi -L
GPU 0: NVIDIA GeForce RTX 3060 (UUID: GPU-5448b7cc-c2e1-c6c5-3ecf-e1115189050b)