• Lei Wang's avatar
    Add Dockerfiles for multiple CUDA versions (#103) · 8d450c34
    Lei Wang authored
    Create Dockerfiles for CUDA versions 11.8, 12.1, 12.3, 12.4, and 12.5, using the latest PyTorch NGC base images. Update docker/README.md to provide clearer instructions for building and running the Docker container with a specific CUDA version.
    8d450c34
Dockerfile.cu121 997 Bytes