update links. Ensure ROCm 6.0

.github/workflows/docker-build.yml

@@ -172,14 +172,14 @@ jobs:
           - "2.3.1"
         rocm:
           - "5.7"
-          - "6.0.2"
+          - "6.0"
         level:
           - "runtime"
         exclude:
           - pytorch: "2.3.1"
             rocm: "5.7"
           - pytorch: "2.2.1"
-            rocm: "6.0.2"
+            rocm: "6.0"
 
     steps:
       -

README.md

@@ -8,7 +8,7 @@ This image provides a great starting point for python development when used stan
 
 ## Documentation
 
-All AI-Dock containers share a common base which is designed to make running on cloud services such as [vast.ai](https://link.ai-dock.org/vast.ai) and [runpod.io](https://link.ai-dock.org/template) as straightforward and user friendly as possible.
+All AI-Dock containers share a common base which is designed to make running on cloud services such as [vast.ai](https://link.ai-dock.org/vast.ai) and [runpod.io](https://link.ai-dock.org/runpod.io) as straightforward and user friendly as possible.
 
 Common features and options are documented in the [base wiki](https://github.com/ai-dock/base-image/wiki) but any additional features unique to this image will be detailed below.
 
@@ -20,25 +20,25 @@ The `:latest` tag points to `:latest-cuda`
 Tags follow these patterns:
 
 ##### _CUDA_
-- `:[pytorch-version]-py[python-version]-cuda-[x.x.x]-base-[ubuntu-version]`
+- `:[pytorch-version]-py[python-version]-v2-cuda-[x.x.x]-base-[ubuntu-version]`
 
-- `:latest-cuda` → `:2.2.1-py3.10-cuda-11.8.0-base-22.04`
+- `:latest-cuda` → `:2.3.1-py3.10-v2-cuda-11.8.0-base-22.04`
 
 ##### _ROCm_
-- `:[pytorch-version]-py[python-version]-rocm-[x.x.x]-runtime-[ubuntu-version]`
+- `:[pytorch-version]-py[python-version]-v2-rocm-[x.x.x]-runtime-[ubuntu-version]`
 
-- `:latest-rocm` → `:2.2.1-py3.10-rocm-5.7-runtime-22.04`
+- `:latest-rocm` → `:2.3.1-py3.10-v2-rocm-6.0-runtime-22.04`
 
 ##### _CPU_
-- `:[pytorch-version]-py[python-version]-ubuntu-[ubuntu-version]`
+- `:[pytorch-version]-py[python-version]-v2-cpu-[ubuntu-version]`
 
-- `:latest-cpu` → `:2.2.1-py3.10-cpu-22.04` 
+- `:latest-cpu` → `:2.3.1-py3.10-v2-cpu-22.04` 
 
 Browse [here](https://github.com/ai-dock/pytorch/pkgs/container/pytorch) for an image suitable for your target environment.
 
 Supported Python versions: `3.12`, `3.11`, `3.10`
 
-Supported Pytorch versions: `2.2.1` `2.1.2`
+Supported Pytorch versions: `2.3.1` `2.2.1`
 
 Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`