• Lei Wang's avatar
    [AMD][Docker] Create Dockerfile for ROCm environment setup (#355) · c58cbfbb
    Lei Wang authored
    * [Add] Create Dockerfile for ROCm environment setup
    
    * Introduced a new Dockerfile for setting up a ROCm environment with PyTorch.
    * Configured the working directory and installed necessary packages including Miniconda, Python, and development tools.
    * Cloned the tilelang repository and executed the ROCm installation script.
    * Set environment variables for compatibility and performance optimization.
    
    * [Remove] Delete Dockerfile for ROCm environment setup
    
    * Removed the Dockerfile used for setting up a ROCm environment with PyTorch.
    * Updated README to reflect changes in Docker image naming conventions for AMD GPU support.
    c58cbfbb
README.md 956 Bytes