Commit Graph

3 Commits

Author SHA1 Message Date
huyyxy aa65200713 feat(docker): update CUDA base image to 12.4.1 for TensorRT support
- Upgrade base image from nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 to nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
- Enable CUDA 12.4 runtime environment
- Ensure TensorRT dependency compatibility
- Validation steps:
  - Verify CUDA version via nvidia-smi after build
  - Test import tensorrt in container without errors

Closes #935
2025-01-26 12:33:50 +08:00
hbybyyang f38f594303 fix docker python version 2025-01-12 15:59:58 +08:00
Dongwon Kim c6b16c06e8 Add dockerfile for python 3.8, cuda 11.8 and torch 2.0.1 version 2024-09-11 11:13:22 +09:00